LostTech.TensorFlow : API Documentation

Type model_pruning

Namespace tensorflow.contrib.model_pruning

Public static methods

object apply_mask(IGraphNodeBase x, VariableScope scope)

object apply_mask(IGraphNodeBase x, string scope)

object apply_mask_dyn(object x, ImplicitContainer<T> scope)

IList<object> get_masked_weights()

object get_masked_weights_dyn()

IList<object> get_masks()

object get_masks_dyn()

HParams get_pruning_hparams()

object get_pruning_hparams_dyn()

IList<object> get_thresholds()

object get_thresholds_dyn()

IList<Tensor> get_weight_sparsity()

object get_weight_sparsity_dyn()

object graph_def_from_checkpoint(object checkpoint_dir, object output_node_names)

object graph_def_from_checkpoint_dyn(object checkpoint_dir, object output_node_names)

Tensor masked_convolution(IGraphNodeBase inputs, int num_outputs, int kernel_size, int stride, string padding, object data_format, int rate, ImplicitContainer<T> activation_fn, PythonFunctionContainer normalizer_fn, object normalizer_params, ImplicitContainer<T> weights_initializer, object weights_regularizer, ImplicitContainer<T> biases_initializer, object biases_regularizer, object reuse, object variables_collections, object outputs_collections, bool trainable, string scope)

object masked_convolution_dyn(object inputs, object num_outputs, object kernel_size, ImplicitContainer<T> stride, ImplicitContainer<T> padding, object data_format, ImplicitContainer<T> rate, ImplicitContainer<T> activation_fn, object normalizer_fn, object normalizer_params, ImplicitContainer<T> weights_initializer, object weights_regularizer, ImplicitContainer<T> biases_initializer, object biases_regularizer, object reuse, object variables_collections, object outputs_collections, ImplicitContainer<T> trainable, object scope)

Tensor masked_fully_connected(IGraphNodeBase inputs, int num_outputs, ImplicitContainer<T> activation_fn, PythonFunctionContainer normalizer_fn, object normalizer_params, ImplicitContainer<T> weights_initializer, object weights_regularizer, ImplicitContainer<T> biases_initializer, object biases_regularizer, object reuse, object variables_collections, object outputs_collections, bool trainable, object scope)

object masked_fully_connected_dyn(object inputs, object num_outputs, ImplicitContainer<T> activation_fn, object normalizer_fn, object normalizer_params, ImplicitContainer<T> weights_initializer, object weights_regularizer, ImplicitContainer<T> biases_initializer, object biases_regularizer, object reuse, object variables_collections, object outputs_collections, ImplicitContainer<T> trainable, object scope)

PythonFunctionContainer strip_pruning_vars_fn(object input_graph_def, IEnumerable<object> output_node_names)

object strip_pruning_vars_fn_dyn(object input_graph_def, object output_node_names)

object train(IGraphNodeBase train_op, object logdir, object mask_update_op, ImplicitContainer<T> train_step_fn, ImplicitContainer<T> train_step_kwargs, int log_every_n_steps, object graph, string master, bool is_chief, object global_step, object number_of_steps, ImplicitContainer<T> init_op, object init_feed_dict, ImplicitContainer<T> local_init_op, PythonFunctionContainer init_fn, ImplicitContainer<T> ready_op, ImplicitContainer<T> summary_op, int save_summaries_secs, ImplicitContainer<T> summary_writer, int startup_delay_steps, object saver, int save_interval_secs, object sync_optimizer, object session_config, object trace_every_n_steps)

object train_dyn(object train_op, object logdir, object mask_update_op, ImplicitContainer<T> train_step_fn, ImplicitContainer<T> train_step_kwargs, ImplicitContainer<T> log_every_n_steps, object graph, ImplicitContainer<T> master, ImplicitContainer<T> is_chief, object global_step, object number_of_steps, ImplicitContainer<T> init_op, object init_feed_dict, ImplicitContainer<T> local_init_op, object init_fn, ImplicitContainer<T> ready_op, ImplicitContainer<T> summary_op, ImplicitContainer<T> save_summaries_secs, ImplicitContainer<T> summary_writer, ImplicitContainer<T> startup_delay_steps, object saver, ImplicitContainer<T> save_interval_secs, object sync_optimizer, object session_config, object trace_every_n_steps)

Public properties

PythonFunctionContainer apply_mask_fn get;

PythonFunctionContainer get_masked_weights_fn get;

PythonFunctionContainer get_masks_fn get;

PythonFunctionContainer get_pruning_hparams_fn get;

PythonFunctionContainer get_thresholds_fn get;

PythonFunctionContainer get_weight_sparsity_fn get;

PythonFunctionContainer get_weights_fn get;

PythonFunctionContainer graph_def_from_checkpoint_fn get;

PythonFunctionContainer masked_convolution_fn get;

PythonFunctionContainer masked_fully_connected_fn get;

PythonFunctionContainer strip_pruning_vars_fn_fn get;

PythonFunctionContainer train_fn get;