LostTech.TensorFlow : API Documentation

Type tpu_ops

Namespace tensorflow.python.tpu.ops.tpu_ops

Methods

Properties

Public static methods

Tensor all_to_all(object x, object concat_dimension, object split_dimension, object split_count, object group_assignment, string name)

object all_to_all_dyn(object x, object concat_dimension, object split_dimension, object split_count, object group_assignment, object name)

Tensor collective_permute(object x, object source_target_pairs, string name)

object collective_permute_dyn(object x, object source_target_pairs, object name)

object enqueue_tpu_embedding_integer_batch(object batch, object device_ordinal, object mode_override, string name)

object enqueue_tpu_embedding_integer_batch_dyn(object batch, object device_ordinal, object mode_override, object name)

object enqueue_tpu_embedding_sparse_batch(object sample_indices, object embedding_indices, object aggregation_weights, object device_ordinal, object combiners, object mode_override, string name)

object enqueue_tpu_embedding_sparse_batch_dyn(object sample_indices, object embedding_indices, object aggregation_weights, object device_ordinal, object combiners, object mode_override, object name)

object enqueue_tpu_embedding_sparse_tensor_batch(IEnumerable<object> sample_indices, IEnumerable<object> embedding_indices, IEnumerable<object> aggregation_weights, IEnumerable<object> table_ids, object device_ordinal, IEnumerable<object> max_sequence_lengths, IEnumerable<object> combiners, object mode_override, string name)

object enqueue_tpu_embedding_sparse_tensor_batch_dyn(object sample_indices, object embedding_indices, object aggregation_weights, object table_ids, object device_ordinal, object max_sequence_lengths, object combiners, object mode_override, object name)

Tensor infeed_dequeue(DType dtype, TensorShape shape, string name)

object infeed_dequeue_dyn(object dtype, object shape, object name)

IList<object> infeed_dequeue_tuple(IEnumerable<DType> dtypes, IEnumerable<TensorShape> shapes, string name)

object infeed_dequeue_tuple_dyn(object dtypes, object shapes, object name)

object send_tpu_embedding_gradients(IEnumerable<IGraphNodeBase> inputs, object config, IEnumerable<object> learning_rates, string name)

object send_tpu_embedding_gradients_dyn(object inputs, object config, object learning_rates, object name)

Public properties

PythonFunctionContainer all_to_all_fn get;

PythonFunctionContainer collective_permute_fn get;

PythonFunctionContainer enqueue_tpu_embedding_integer_batch_fn get;

PythonFunctionContainer enqueue_tpu_embedding_sparse_batch_fn get;

PythonFunctionContainer enqueue_tpu_embedding_sparse_tensor_batch_fn get;

PythonFunctionContainer infeed_dequeue_fn get;

PythonFunctionContainer infeed_dequeue_tuple_fn get;

PythonFunctionContainer send_tpu_embedding_gradients_fn get;