LostTech.TensorFlow : API Documentation

Type tpu_embedding_gradient

Namespace tensorflow.python.tpu.tpu_embedding_gradient

Public static methods

ValueTuple<PythonClassContainer, object> create_dummy_table_variables(object tpu_embedding)

object create_dummy_table_variables_dyn(object tpu_embedding)

PythonClassContainer get_gradients_through_compute_gradients(object optimizer, object loss, object activations)

object get_gradients_through_compute_gradients_dyn(object optimizer, object loss, object activations)

PythonClassContainer get_gradients_through_dummy_table_variables(object tpu_embedding)

object get_gradients_through_dummy_table_variables_dyn(object tpu_embedding)

PythonClassContainer hook_dummy_table_variables_to_activations(object tpu_embedding, object activations, object dummy_table_variables)

object hook_dummy_table_variables_to_activations_dyn(object tpu_embedding, object activations, object dummy_table_variables)

Public properties

PythonFunctionContainer create_dummy_table_variables_fn get;

PythonFunctionContainer get_gradients_through_compute_gradients_fn get;

PythonFunctionContainer get_gradients_through_dummy_table_variables_fn get;

PythonFunctionContainer hook_dummy_table_variables_to_activations_fn get;