LostTech.TensorFlow : API Documentation

Type TPUEmbedding

Namespace tensorflow.python.tpu.tpu_embedding

Parent PythonObjectContainer

Interfaces ITPUEmbedding

Methods

Properties

Public instance methods

object create_variables_and_ops(object embedding_variable_name_by_table, object slot_variable_names_by_table)

object create_variables_and_ops_dyn(object embedding_variable_name_by_table, object slot_variable_names_by_table)

object generate_send_gradients_op(object feature_to_gradient_dict, object learning_rates)

object generate_send_gradients_op_dyn(object feature_to_gradient_dict, object learning_rates)

PythonClassContainer get_activations()

object get_activations_dyn()

Public static methods

TPUEmbedding NewDyn(object table_to_config_dict, object feature_to_config_dict, object batch_size, object mode, object master, object optimization_parameters, object cluster_def, ImplicitContainer<T> pipeline_execution_with_tensor_core, ImplicitContainer<T> partition_strategy, object device_config)

Public properties

object batch_size_per_core get;

object batch_size_per_core_dyn get;

object config_proto get;

object config_proto_dyn get;

object feature_to_config_dict get;

object feature_to_config_dict_dyn get;

object hosts get;

object hosts_dyn get;

object num_cores get;

object num_cores_dyn get;

object num_cores_per_host get;

object num_cores_per_host_dyn get;

StochasticGradientDescentParameters optimization_parameters get;

object optimization_parameters_dyn get;

object PythonObject get;

object table_to_config_dict get;

object table_to_config_dict_dyn get;

object table_to_features_dict get;

object table_to_features_dict_dyn get;