LostTech.TensorFlow : API Documentation

Type gen_tpu_ops

Namespace tensorflow.python.ops.gen_tpu_ops

Methods

Properties

Public static methods

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

object all_to_all_eager_fallback(IGraphNodeBase input, IGraphNodeBase group_assignment, int concat_dimension, int split_dimension, int split_count, string name, Context ctx)

object all_to_all_eager_fallback_dyn(object input, object group_assignment, object concat_dimension, object split_dimension, object split_count, object name, object ctx)

Tensor collective_permute(IGraphNodeBase input, IGraphNodeBase source_target_pairs, string name)

object collective_permute_eager_fallback(IGraphNodeBase input, IGraphNodeBase source_target_pairs, string name, Context ctx)

object collective_permute_eager_fallback_dyn(object input, object source_target_pairs, object name, object ctx)

Tensor configure_distributed_tpu(string embedding_config, string tpu_embedding_config, bool is_global_init, string name)

object configure_distributed_tpu_dyn(ImplicitContainer<T> embedding_config, ImplicitContainer<T> tpu_embedding_config, ImplicitContainer<T> is_global_init, object name)

object configure_distributed_tpu_eager_fallback(string embedding_config, string tpu_embedding_config, bool is_global_init, string name, Context ctx)

object configure_distributed_tpu_eager_fallback(string embedding_config, Byte[] tpu_embedding_config, bool is_global_init, string name, Context ctx)

object configure_distributed_tpu_eager_fallback(Byte[] embedding_config, string tpu_embedding_config, bool is_global_init, string name, Context ctx)

object configure_distributed_tpu_eager_fallback(Byte[] embedding_config, Byte[] tpu_embedding_config, bool is_global_init, string name, Context ctx)

object configure_distributed_tpu_eager_fallback_dyn(ImplicitContainer<T> embedding_config, ImplicitContainer<T> tpu_embedding_config, ImplicitContainer<T> is_global_init, object name, object ctx)

object configure_tpu_embedding(object config, string name)

object configure_tpu_embedding_dyn(object config, object name)

object configure_tpu_embedding_eager_fallback(Byte[] config, string name, Context ctx)

object configure_tpu_embedding_eager_fallback_dyn(object config, object name, object ctx)

Tensor cross_replica_sum(IGraphNodeBase input, IGraphNodeBase group_assignment, string name)

object cross_replica_sum_dyn(object input, object group_assignment, object name)

Sum the input tensor across replicas according to group_assignment.
Parameters
object input
object group_assignment
Optional 2d int32 lists with shape [num_groups, num_replicas_per_group]. `group_assignment[i]` represents the replica ids in the ith subgroup.
object name
Optional op name.
Returns
object
A `Tensor` which is summed across replicas.

object cross_replica_sum_eager_fallback(IGraphNodeBase input, IGraphNodeBase group_assignment, string name, Context ctx)

object cross_replica_sum_eager_fallback_dyn(object input, object group_assignment, object name, object ctx)

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

object enqueue_tpu_embedding_integer_batch_dyn(object batch, object mode_override, ImplicitContainer<T> device_ordinal, object name)

object enqueue_tpu_embedding_integer_batch_eager_fallback(object batch, IGraphNodeBase mode_override, int device_ordinal, string name, Context ctx)

object enqueue_tpu_embedding_integer_batch_eager_fallback_dyn(object batch, object mode_override, ImplicitContainer<T> device_ordinal, object name, object ctx)

object enqueue_tpu_embedding_sparse_batch(object sample_indices, object embedding_indices, object aggregation_weights, IGraphNodeBase mode_override, int device_ordinal, ImplicitContainer<T> combiners, string name)

object enqueue_tpu_embedding_sparse_batch_dyn(object sample_indices, object embedding_indices, object aggregation_weights, object mode_override, ImplicitContainer<T> device_ordinal, ImplicitContainer<T> combiners, object name)

object enqueue_tpu_embedding_sparse_batch_eager_fallback(object sample_indices, object embedding_indices, object aggregation_weights, IGraphNodeBase mode_override, int device_ordinal, ImplicitContainer<T> combiners, string name, Context ctx)

object enqueue_tpu_embedding_sparse_batch_eager_fallback_dyn(object sample_indices, object embedding_indices, object aggregation_weights, object mode_override, ImplicitContainer<T> device_ordinal, ImplicitContainer<T> combiners, object name, object ctx)

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

object enqueue_tpu_embedding_sparse_tensor_batch_dyn(object sample_indices, object embedding_indices, object aggregation_weights, object mode_override, object table_ids, ImplicitContainer<T> device_ordinal, ImplicitContainer<T> combiners, ImplicitContainer<T> max_sequence_lengths, object name)

object enqueue_tpu_embedding_sparse_tensor_batch_eager_fallback(IEnumerable<object> sample_indices, IEnumerable<object> embedding_indices, IEnumerable<object> aggregation_weights, IGraphNodeBase mode_override, IEnumerable<int> table_ids, int device_ordinal, ImplicitContainer<T> combiners, ImplicitContainer<T> max_sequence_lengths, string name, Context ctx)

object enqueue_tpu_embedding_sparse_tensor_batch_eager_fallback_dyn(object sample_indices, object embedding_indices, object aggregation_weights, object mode_override, object table_ids, ImplicitContainer<T> device_ordinal, ImplicitContainer<T> combiners, ImplicitContainer<T> max_sequence_lengths, object name, object ctx)

object infeed_dequeue_eager_fallback(int dtype, IEnumerable<Nullable<int>> shape, string name, Context ctx)

object infeed_dequeue_eager_fallback_dyn(object dtype, object shape, object name, object ctx)

IList<object> infeed_dequeue_tuple_eager_fallback(IEnumerable<object> dtypes, IEnumerable<object> shapes, string name, Context ctx)

object infeed_dequeue_tuple_eager_fallback_dyn(object dtypes, object shapes, object name, object ctx)

object infeed_enqueue(IGraphNodeBase input, ImplicitContainer<T> shape, ImplicitContainer<T> layout, int device_ordinal, string name)

object infeed_enqueue_dyn(object input, ImplicitContainer<T> shape, ImplicitContainer<T> layout, ImplicitContainer<T> device_ordinal, object name)

object infeed_enqueue_eager_fallback(IGraphNodeBase input, ImplicitContainer<T> shape, ImplicitContainer<T> layout, int device_ordinal, string name, Context ctx)

object infeed_enqueue_eager_fallback_dyn(object input, ImplicitContainer<T> shape, ImplicitContainer<T> layout, ImplicitContainer<T> device_ordinal, object name, object ctx)

object infeed_enqueue_prelinearized_buffer(IGraphNodeBase input, int device_ordinal, string name)

object infeed_enqueue_prelinearized_buffer_dyn(object input, ImplicitContainer<T> device_ordinal, object name)

object infeed_enqueue_prelinearized_buffer_eager_fallback(IGraphNodeBase input, int device_ordinal, string name, Context ctx)

object infeed_enqueue_prelinearized_buffer_eager_fallback_dyn(object input, ImplicitContainer<T> device_ordinal, object name, object ctx)

object infeed_enqueue_tuple(IEnumerable<IGraphNodeBase> inputs, IEnumerable<object> shapes, ImplicitContainer<T> layouts, int device_ordinal, int name)

object infeed_enqueue_tuple(IEnumerable<IGraphNodeBase> inputs, IEnumerable<object> shapes, ImplicitContainer<T> layouts, int device_ordinal, string name)

object infeed_enqueue_tuple_dyn(object inputs, object shapes, ImplicitContainer<T> layouts, ImplicitContainer<T> device_ordinal, object name)

object infeed_enqueue_tuple_eager_fallback(IEnumerable<IGraphNodeBase> inputs, IEnumerable<object> shapes, ImplicitContainer<T> layouts, int device_ordinal, string name, Context ctx)

object infeed_enqueue_tuple_eager_fallback(IEnumerable<IGraphNodeBase> inputs, IEnumerable<object> shapes, ImplicitContainer<T> layouts, int device_ordinal, int name, Context ctx)

object infeed_enqueue_tuple_eager_fallback_dyn(object inputs, object shapes, ImplicitContainer<T> layouts, ImplicitContainer<T> device_ordinal, object name, object ctx)

object load_tpu_embedding_adadelta_parameters(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase updates, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_adadelta_parameters_dyn(object parameters, object accumulators, object updates, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_adadelta_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase updates, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_adadelta_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase updates, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_adadelta_parameters_eager_fallback_dyn(object parameters, object accumulators, object updates, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object load_tpu_embedding_adadelta_parameters_grad_accum_debug(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase updates, IGraphNodeBase gradient_accumulators, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_adadelta_parameters_grad_accum_debug_dyn(object parameters, object accumulators, object updates, object gradient_accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_adadelta_parameters_grad_accum_debug_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase updates, IGraphNodeBase gradient_accumulators, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_adadelta_parameters_grad_accum_debug_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase updates, IGraphNodeBase gradient_accumulators, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_adadelta_parameters_grad_accum_debug_eager_fallback_dyn(object parameters, object accumulators, object updates, object gradient_accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object load_tpu_embedding_adagrad_parameters(IGraphNodeBase parameters, IGraphNodeBase accumulators, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_adagrad_parameters_dyn(object parameters, object accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_adagrad_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_adagrad_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_adagrad_parameters_eager_fallback_dyn(object parameters, object accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object load_tpu_embedding_adagrad_parameters_grad_accum_debug(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase gradient_accumulators, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_adagrad_parameters_grad_accum_debug_dyn(object parameters, object accumulators, object gradient_accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_adagrad_parameters_grad_accum_debug_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase gradient_accumulators, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_adagrad_parameters_grad_accum_debug_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase gradient_accumulators, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_adagrad_parameters_grad_accum_debug_eager_fallback_dyn(object parameters, object accumulators, object gradient_accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object load_tpu_embedding_adam_parameters(IGraphNodeBase parameters, IGraphNodeBase momenta, IGraphNodeBase velocities, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_adam_parameters_dyn(object parameters, object momenta, object velocities, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_adam_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase momenta, IGraphNodeBase velocities, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_adam_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase momenta, IGraphNodeBase velocities, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_adam_parameters_eager_fallback_dyn(object parameters, object momenta, object velocities, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object load_tpu_embedding_adam_parameters_grad_accum_debug(IGraphNodeBase parameters, IGraphNodeBase momenta, IGraphNodeBase velocities, IGraphNodeBase gradient_accumulators, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_adam_parameters_grad_accum_debug_dyn(object parameters, object momenta, object velocities, object gradient_accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_adam_parameters_grad_accum_debug_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase momenta, IGraphNodeBase velocities, IGraphNodeBase gradient_accumulators, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_adam_parameters_grad_accum_debug_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase momenta, IGraphNodeBase velocities, IGraphNodeBase gradient_accumulators, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_adam_parameters_grad_accum_debug_eager_fallback_dyn(object parameters, object momenta, object velocities, object gradient_accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object load_tpu_embedding_centered_rms_prop_parameters(IGraphNodeBase parameters, IGraphNodeBase ms, IGraphNodeBase mom, IGraphNodeBase mg, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_centered_rms_prop_parameters_dyn(object parameters, object ms, object mom, object mg, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_centered_rms_prop_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase ms, IGraphNodeBase mom, IGraphNodeBase mg, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_centered_rms_prop_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase ms, IGraphNodeBase mom, IGraphNodeBase mg, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_centered_rms_prop_parameters_eager_fallback_dyn(object parameters, object ms, object mom, object mg, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object load_tpu_embedding_ftrl_parameters(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase linears, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_ftrl_parameters_dyn(object parameters, object accumulators, object linears, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_ftrl_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase linears, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_ftrl_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase linears, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_ftrl_parameters_eager_fallback_dyn(object parameters, object accumulators, object linears, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object load_tpu_embedding_ftrl_parameters_grad_accum_debug(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase linears, IGraphNodeBase gradient_accumulators, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_ftrl_parameters_grad_accum_debug_dyn(object parameters, object accumulators, object linears, object gradient_accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_ftrl_parameters_grad_accum_debug_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase linears, IGraphNodeBase gradient_accumulators, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_ftrl_parameters_grad_accum_debug_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase linears, IGraphNodeBase gradient_accumulators, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_ftrl_parameters_grad_accum_debug_eager_fallback_dyn(object parameters, object accumulators, object linears, object gradient_accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object load_tpu_embedding_mdl_adagrad_light_parameters(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase weights, IGraphNodeBase benefits, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_mdl_adagrad_light_parameters_dyn(object parameters, object accumulators, object weights, object benefits, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_mdl_adagrad_light_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase weights, IGraphNodeBase benefits, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_mdl_adagrad_light_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase weights, IGraphNodeBase benefits, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_mdl_adagrad_light_parameters_eager_fallback_dyn(object parameters, object accumulators, object weights, object benefits, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object load_tpu_embedding_momentum_parameters(IGraphNodeBase parameters, IGraphNodeBase momenta, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_momentum_parameters_dyn(object parameters, object momenta, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_momentum_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase momenta, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_momentum_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase momenta, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_momentum_parameters_eager_fallback_dyn(object parameters, object momenta, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object load_tpu_embedding_momentum_parameters_grad_accum_debug(IGraphNodeBase parameters, IGraphNodeBase momenta, IGraphNodeBase gradient_accumulators, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_momentum_parameters_grad_accum_debug_dyn(object parameters, object momenta, object gradient_accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_momentum_parameters_grad_accum_debug_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase momenta, IGraphNodeBase gradient_accumulators, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_momentum_parameters_grad_accum_debug_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase momenta, IGraphNodeBase gradient_accumulators, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_momentum_parameters_grad_accum_debug_eager_fallback_dyn(object parameters, object momenta, object gradient_accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object load_tpu_embedding_proximal_adagrad_parameters(IGraphNodeBase parameters, IGraphNodeBase accumulators, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_proximal_adagrad_parameters_dyn(object parameters, object accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_proximal_adagrad_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_proximal_adagrad_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_proximal_adagrad_parameters_eager_fallback_dyn(object parameters, object accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object load_tpu_embedding_proximal_adagrad_parameters_grad_accum_debug(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase gradient_accumulators, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_proximal_adagrad_parameters_grad_accum_debug_dyn(object parameters, object accumulators, object gradient_accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_proximal_adagrad_parameters_grad_accum_debug_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase gradient_accumulators, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_proximal_adagrad_parameters_grad_accum_debug_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase accumulators, IGraphNodeBase gradient_accumulators, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_proximal_adagrad_parameters_grad_accum_debug_eager_fallback_dyn(object parameters, object accumulators, object gradient_accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object load_tpu_embedding_rms_prop_parameters(IGraphNodeBase parameters, IGraphNodeBase ms, IGraphNodeBase mom, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_rms_prop_parameters_dyn(object parameters, object ms, object mom, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_rms_prop_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase ms, IGraphNodeBase mom, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_rms_prop_parameters_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase ms, IGraphNodeBase mom, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_rms_prop_parameters_eager_fallback_dyn(object parameters, object ms, object mom, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object load_tpu_embedding_rms_prop_parameters_grad_accum_debug(IGraphNodeBase parameters, IGraphNodeBase ms, IGraphNodeBase mom, IGraphNodeBase gradient_accumulators, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_rms_prop_parameters_grad_accum_debug_dyn(object parameters, object ms, object mom, object gradient_accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_rms_prop_parameters_grad_accum_debug_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase ms, IGraphNodeBase mom, IGraphNodeBase gradient_accumulators, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_rms_prop_parameters_grad_accum_debug_eager_fallback(IGraphNodeBase parameters, IGraphNodeBase ms, IGraphNodeBase mom, IGraphNodeBase gradient_accumulators, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_rms_prop_parameters_grad_accum_debug_eager_fallback_dyn(object parameters, object ms, object mom, object gradient_accumulators, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object load_tpu_embedding_stochastic_gradient_descent_parameters(IGraphNodeBase parameters, object num_shards, object shard_id, int table_id, string table_name, string name)

object load_tpu_embedding_stochastic_gradient_descent_parameters_dyn(object parameters, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object load_tpu_embedding_stochastic_gradient_descent_parameters_eager_fallback(IGraphNodeBase parameters, int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object load_tpu_embedding_stochastic_gradient_descent_parameters_eager_fallback(IGraphNodeBase parameters, int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object load_tpu_embedding_stochastic_gradient_descent_parameters_eager_fallback_dyn(object parameters, object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

Tensor outfeed_dequeue(DType dtype, TensorShape shape, int device_ordinal, string name)

object outfeed_dequeue_dyn(object dtype, object shape, ImplicitContainer<T> device_ordinal, object name)

object outfeed_dequeue_eager_fallback(int dtype, IEnumerable<Nullable<int>> shape, int device_ordinal, string name, Context ctx)

object outfeed_dequeue_eager_fallback_dyn(object dtype, object shape, ImplicitContainer<T> device_ordinal, object name, object ctx)

IList<object> outfeed_dequeue_tuple(IEnumerable<DType> dtypes, IEnumerable<TensorShape> shapes, int device_ordinal, string name)

object outfeed_dequeue_tuple_dyn(object dtypes, object shapes, ImplicitContainer<T> device_ordinal, object name)

IList<object> outfeed_dequeue_tuple_eager_fallback(IEnumerable<object> dtypes, IEnumerable<object> shapes, int device_ordinal, string name, Context ctx)

object outfeed_dequeue_tuple_eager_fallback_dyn(object dtypes, object shapes, ImplicitContainer<T> device_ordinal, object name, object ctx)

object outfeed_enqueue(IGraphNodeBase input, string name)

object outfeed_enqueue_dyn(object input, object name)

object outfeed_enqueue_eager_fallback(IGraphNodeBase input, string name, Context ctx)

object outfeed_enqueue_eager_fallback_dyn(object input, object name, object ctx)

object outfeed_enqueue_tuple(IEnumerable<object> inputs, string name)

object outfeed_enqueue_tuple_dyn(object inputs, object name)

object outfeed_enqueue_tuple_eager_fallback(IEnumerable<object> inputs, string name, Context ctx)

object outfeed_enqueue_tuple_eager_fallback_dyn(object inputs, object name, object ctx)

Tensor prelinearize(IGraphNodeBase input, ImplicitContainer<T> shape, ImplicitContainer<T> layout, string name)

object prelinearize_dyn(object input, ImplicitContainer<T> shape, ImplicitContainer<T> layout, object name)

object prelinearize_eager_fallback(IGraphNodeBase input, ImplicitContainer<T> shape, ImplicitContainer<T> layout, string name, Context ctx)

object prelinearize_eager_fallback_dyn(object input, ImplicitContainer<T> shape, ImplicitContainer<T> layout, object name, object ctx)

Tensor prelinearize_tuple(object inputs, object shapes, ImplicitContainer<T> layouts, string name)

object prelinearize_tuple_dyn(object inputs, object shapes, ImplicitContainer<T> layouts, object name)

object prelinearize_tuple_eager_fallback(object inputs, IEnumerable<object> shapes, ImplicitContainer<T> layouts, string name, Context ctx)

object prelinearize_tuple_eager_fallback_dyn(object inputs, object shapes, ImplicitContainer<T> layouts, object name, object ctx)

IList<object> recv_tpu_embedding_activations(int num_outputs, object config, string name)

object recv_tpu_embedding_activations_dyn(object num_outputs, object config, object name)

IList<object> recv_tpu_embedding_activations_eager_fallback(int num_outputs, Byte[] config, string name, Context ctx)

object recv_tpu_embedding_activations_eager_fallback_dyn(object num_outputs, object config, object name, object ctx)

object retrieve_tpu_embedding_adadelta_parameters(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_adadelta_parameters_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_adadelta_parameters_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_adadelta_parameters_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_adadelta_parameters_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object retrieve_tpu_embedding_adadelta_parameters_grad_accum_debug(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_adadelta_parameters_grad_accum_debug_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_adadelta_parameters_grad_accum_debug_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_adadelta_parameters_grad_accum_debug_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_adadelta_parameters_grad_accum_debug_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object retrieve_tpu_embedding_adagrad_parameters(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_adagrad_parameters_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_adagrad_parameters_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_adagrad_parameters_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_adagrad_parameters_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object retrieve_tpu_embedding_adagrad_parameters_grad_accum_debug(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_adagrad_parameters_grad_accum_debug_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_adagrad_parameters_grad_accum_debug_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_adagrad_parameters_grad_accum_debug_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_adagrad_parameters_grad_accum_debug_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object retrieve_tpu_embedding_adam_parameters(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_adam_parameters_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_adam_parameters_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_adam_parameters_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_adam_parameters_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object retrieve_tpu_embedding_adam_parameters_grad_accum_debug(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_adam_parameters_grad_accum_debug_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_adam_parameters_grad_accum_debug_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_adam_parameters_grad_accum_debug_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_adam_parameters_grad_accum_debug_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object retrieve_tpu_embedding_centered_rms_prop_parameters(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_centered_rms_prop_parameters_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_centered_rms_prop_parameters_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_centered_rms_prop_parameters_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_centered_rms_prop_parameters_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object retrieve_tpu_embedding_ftrl_parameters(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_ftrl_parameters_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_ftrl_parameters_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_ftrl_parameters_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_ftrl_parameters_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object retrieve_tpu_embedding_ftrl_parameters_grad_accum_debug(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_ftrl_parameters_grad_accum_debug_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_ftrl_parameters_grad_accum_debug_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_ftrl_parameters_grad_accum_debug_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_ftrl_parameters_grad_accum_debug_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object retrieve_tpu_embedding_mdl_adagrad_light_parameters(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_mdl_adagrad_light_parameters_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_mdl_adagrad_light_parameters_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_mdl_adagrad_light_parameters_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_mdl_adagrad_light_parameters_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object retrieve_tpu_embedding_momentum_parameters(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_momentum_parameters_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_momentum_parameters_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_momentum_parameters_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_momentum_parameters_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object retrieve_tpu_embedding_momentum_parameters_grad_accum_debug(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_momentum_parameters_grad_accum_debug_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_momentum_parameters_grad_accum_debug_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_momentum_parameters_grad_accum_debug_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_momentum_parameters_grad_accum_debug_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object retrieve_tpu_embedding_proximal_adagrad_parameters(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_proximal_adagrad_parameters_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_proximal_adagrad_parameters_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_proximal_adagrad_parameters_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_proximal_adagrad_parameters_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object retrieve_tpu_embedding_proximal_adagrad_parameters_grad_accum_debug(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_proximal_adagrad_parameters_grad_accum_debug_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_proximal_adagrad_parameters_grad_accum_debug_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_proximal_adagrad_parameters_grad_accum_debug_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_proximal_adagrad_parameters_grad_accum_debug_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object retrieve_tpu_embedding_rms_prop_parameters(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_rms_prop_parameters_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_rms_prop_parameters_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_rms_prop_parameters_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_rms_prop_parameters_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

object retrieve_tpu_embedding_rms_prop_parameters_grad_accum_debug(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_rms_prop_parameters_grad_accum_debug_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_rms_prop_parameters_grad_accum_debug_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_rms_prop_parameters_grad_accum_debug_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_rms_prop_parameters_grad_accum_debug_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

Tensor retrieve_tpu_embedding_stochastic_gradient_descent_parameters(object num_shards, object shard_id, int table_id, string table_name, string name)

object retrieve_tpu_embedding_stochastic_gradient_descent_parameters_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name)

object retrieve_tpu_embedding_stochastic_gradient_descent_parameters_eager_fallback(int num_shards, int shard_id, int table_id, Byte[] table_name, string name, Context ctx)

object retrieve_tpu_embedding_stochastic_gradient_descent_parameters_eager_fallback(int num_shards, int shard_id, int table_id, string table_name, string name, Context ctx)

object retrieve_tpu_embedding_stochastic_gradient_descent_parameters_eager_fallback_dyn(object num_shards, object shard_id, ImplicitContainer<T> table_id, ImplicitContainer<T> table_name, object name, object ctx)

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

object send_tpu_embedding_gradients_eager_fallback(IEnumerable<IGraphNodeBase> inputs, IEnumerable<object> learning_rates, Byte[] config, string name, Context ctx)

object send_tpu_embedding_gradients_eager_fallback_dyn(object inputs, object learning_rates, object config, object name, object ctx)

object shutdown_distributed_tpu(string name)

object shutdown_distributed_tpu_dyn(object name)

object shutdown_distributed_tpu_eager_fallback(string name, Context ctx)

object shutdown_distributed_tpu_eager_fallback_dyn(object name, object ctx)

Tensor tpu_compilation_result(string name)

object tpu_compilation_result_dyn(object name)

object tpu_compilation_result_eager_fallback(string name, Context ctx)

object tpu_compilation_result_eager_fallback_dyn(object name, object ctx)

Tensor tpu_embedding_activations(IGraphNodeBase embedding_variable, IGraphNodeBase sliced_activations, int table_id, object lookup_id, string name)

object tpu_embedding_activations_dyn(object embedding_variable, object sliced_activations, object table_id, object lookup_id, object name)

object tpu_embedding_activations_eager_fallback(IGraphNodeBase embedding_variable, IGraphNodeBase sliced_activations, int table_id, int lookup_id, string name, Context ctx)

object tpu_embedding_activations_eager_fallback_dyn(object embedding_variable, object sliced_activations, object table_id, object lookup_id, object name, object ctx)

Tensor tpu_ordinal_selector(string name)

object tpu_ordinal_selector_dyn(object name)

object tpu_ordinal_selector_eager_fallback(string name, Context ctx)

object tpu_ordinal_selector_eager_fallback_dyn(object name, object ctx)

IList<object> tpu_partitioned_call(IEnumerable<object> args, IGraphNodeBase device_ordinal, IEnumerable<object> Tout, _DefinedFunction f, string name)

IList<object> tpu_partitioned_call(IEnumerable<object> args, IGraphNodeBase device_ordinal, IEnumerable<object> Tout, _OverloadedFunction f, string name)

IList<object> tpu_partitioned_call(IEnumerable<object> args, IGraphNodeBase device_ordinal, IEnumerable<object> Tout, object f, string name)

object tpu_partitioned_call_dyn(object args, object device_ordinal, object Tout, object f, object name)

IList<object> tpu_partitioned_call_eager_fallback(IEnumerable<object> args, IGraphNodeBase device_ordinal, IEnumerable<object> Tout, object f, string name, Context ctx)

IList<object> tpu_partitioned_call_eager_fallback(IEnumerable<object> args, IGraphNodeBase device_ordinal, IEnumerable<object> Tout, _OverloadedFunction f, string name, Context ctx)

IList<object> tpu_partitioned_call_eager_fallback(IEnumerable<object> args, IGraphNodeBase device_ordinal, IEnumerable<object> Tout, _DefinedFunction f, string name, Context ctx)

object tpu_partitioned_call_eager_fallback_dyn(object args, object device_ordinal, object Tout, object f, object name, object ctx)

object tpu_replicate_metadata(int num_replicas, int num_cores_per_replica, string topology, bool use_tpu, ImplicitContainer<T> device_assignment, ImplicitContainer<T> computation_shape, ImplicitContainer<T> host_compute_core, ImplicitContainer<T> padding_map, string step_marker_location, bool allow_soft_placement, string name)

object tpu_replicate_metadata_dyn(object num_replicas, ImplicitContainer<T> num_cores_per_replica, ImplicitContainer<T> topology, ImplicitContainer<T> use_tpu, ImplicitContainer<T> device_assignment, ImplicitContainer<T> computation_shape, ImplicitContainer<T> host_compute_core, ImplicitContainer<T> padding_map, ImplicitContainer<T> step_marker_location, ImplicitContainer<T> allow_soft_placement, object name)

object tpu_replicate_metadata_eager_fallback(int num_replicas, int num_cores_per_replica, Byte[] topology, bool use_tpu, ImplicitContainer<T> device_assignment, ImplicitContainer<T> computation_shape, ImplicitContainer<T> host_compute_core, ImplicitContainer<T> padding_map, Byte[] step_marker_location, bool allow_soft_placement, string name, Context ctx)

object tpu_replicate_metadata_eager_fallback(int num_replicas, int num_cores_per_replica, Byte[] topology, bool use_tpu, ImplicitContainer<T> device_assignment, ImplicitContainer<T> computation_shape, ImplicitContainer<T> host_compute_core, ImplicitContainer<T> padding_map, string step_marker_location, bool allow_soft_placement, string name, Context ctx)

object tpu_replicate_metadata_eager_fallback(int num_replicas, int num_cores_per_replica, string topology, bool use_tpu, ImplicitContainer<T> device_assignment, ImplicitContainer<T> computation_shape, ImplicitContainer<T> host_compute_core, ImplicitContainer<T> padding_map, Byte[] step_marker_location, bool allow_soft_placement, string name, Context ctx)

object tpu_replicate_metadata_eager_fallback(int num_replicas, int num_cores_per_replica, string topology, bool use_tpu, ImplicitContainer<T> device_assignment, ImplicitContainer<T> computation_shape, ImplicitContainer<T> host_compute_core, ImplicitContainer<T> padding_map, string step_marker_location, bool allow_soft_placement, string name, Context ctx)

object tpu_replicate_metadata_eager_fallback_dyn(object num_replicas, ImplicitContainer<T> num_cores_per_replica, ImplicitContainer<T> topology, ImplicitContainer<T> use_tpu, ImplicitContainer<T> device_assignment, ImplicitContainer<T> computation_shape, ImplicitContainer<T> host_compute_core, ImplicitContainer<T> padding_map, ImplicitContainer<T> step_marker_location, ImplicitContainer<T> allow_soft_placement, object name, object ctx)

Tensor tpu_replicated_input(IEnumerable<int> inputs, string name)

Tensor tpu_replicated_input(IEnumerable<int> inputs, int name)

Tensor tpu_replicated_input(IEnumerable<int> inputs, IGraphNodeBase name)

object tpu_replicated_input_dyn(object inputs, object name)

object tpu_replicated_input_eager_fallback(IEnumerable<int> inputs, string name, Context ctx)

object tpu_replicated_input_eager_fallback(IEnumerable<int> inputs, int name, Context ctx)

object tpu_replicated_input_eager_fallback_dyn(object inputs, object name, object ctx)

IList<object> tpu_replicated_output(IGraphNodeBase input, int num_replicas, IGraphNodeBase name)

IList<object> tpu_replicated_output(IGraphNodeBase input, int num_replicas, int name)

object tpu_replicated_output_dyn(object input, object num_replicas, object name)

IList<object> tpu_replicated_output_eager_fallback(IGraphNodeBase input, int num_replicas, int name, Context ctx)

IList<object> tpu_replicated_output_eager_fallback(IGraphNodeBase input, int num_replicas, IGraphNodeBase name, Context ctx)

object tpu_replicated_output_eager_fallback_dyn(object input, object num_replicas, object name, object ctx)

Tensor worker_heartbeat(IGraphNodeBase request, string name)

object worker_heartbeat_dyn(object request, object name)

object worker_heartbeat_eager_fallback(IGraphNodeBase request, string name, Context ctx)

object worker_heartbeat_eager_fallback_dyn(object request, object name, object ctx)

Public properties

PythonFunctionContainer all_to_all_eager_fallback_fn get;

PythonFunctionContainer all_to_all_fn get;

PythonFunctionContainer collective_permute_eager_fallback_fn get;

PythonFunctionContainer collective_permute_fn get;

PythonFunctionContainer configure_distributed_tpu_eager_fallback_fn get;

PythonFunctionContainer configure_distributed_tpu_fn get;

PythonFunctionContainer configure_tpu_embedding_eager_fallback_fn get;

PythonFunctionContainer configure_tpu_embedding_fn get;

PythonFunctionContainer cross_replica_sum_eager_fallback_fn get;

PythonFunctionContainer cross_replica_sum_fn get;

PythonFunctionContainer enqueue_tpu_embedding_integer_batch_eager_fallback_fn get;

PythonFunctionContainer enqueue_tpu_embedding_integer_batch_fn get;

PythonFunctionContainer enqueue_tpu_embedding_sparse_batch_eager_fallback_fn get;

PythonFunctionContainer enqueue_tpu_embedding_sparse_batch_fn get;

PythonFunctionContainer enqueue_tpu_embedding_sparse_tensor_batch_eager_fallback_fn get;

PythonFunctionContainer enqueue_tpu_embedding_sparse_tensor_batch_fn get;

PythonFunctionContainer infeed_dequeue_eager_fallback_fn get;

PythonFunctionContainer infeed_dequeue_fn get;

PythonFunctionContainer infeed_dequeue_tuple_eager_fallback_fn get;

PythonFunctionContainer infeed_dequeue_tuple_fn get;

PythonFunctionContainer infeed_enqueue_eager_fallback_fn get;

PythonFunctionContainer infeed_enqueue_fn get;

PythonFunctionContainer infeed_enqueue_prelinearized_buffer_eager_fallback_fn get;

PythonFunctionContainer infeed_enqueue_prelinearized_buffer_fn get;

PythonFunctionContainer infeed_enqueue_tuple_eager_fallback_fn get;

PythonFunctionContainer infeed_enqueue_tuple_fn get;

PythonFunctionContainer load_tpu_embedding_adadelta_parameters_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_adadelta_parameters_fn get;

PythonFunctionContainer load_tpu_embedding_adadelta_parameters_grad_accum_debug_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_adadelta_parameters_grad_accum_debug_fn get;

PythonFunctionContainer load_tpu_embedding_adagrad_parameters_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_adagrad_parameters_fn get;

PythonFunctionContainer load_tpu_embedding_adagrad_parameters_grad_accum_debug_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_adagrad_parameters_grad_accum_debug_fn get;

PythonFunctionContainer load_tpu_embedding_adam_parameters_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_adam_parameters_fn get;

PythonFunctionContainer load_tpu_embedding_adam_parameters_grad_accum_debug_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_adam_parameters_grad_accum_debug_fn get;

PythonFunctionContainer load_tpu_embedding_centered_rms_prop_parameters_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_centered_rms_prop_parameters_fn get;

PythonFunctionContainer load_tpu_embedding_ftrl_parameters_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_ftrl_parameters_fn get;

PythonFunctionContainer load_tpu_embedding_ftrl_parameters_grad_accum_debug_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_ftrl_parameters_grad_accum_debug_fn get;

PythonFunctionContainer load_tpu_embedding_mdl_adagrad_light_parameters_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_mdl_adagrad_light_parameters_fn get;

PythonFunctionContainer load_tpu_embedding_momentum_parameters_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_momentum_parameters_fn get;

PythonFunctionContainer load_tpu_embedding_momentum_parameters_grad_accum_debug_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_momentum_parameters_grad_accum_debug_fn get;

PythonFunctionContainer load_tpu_embedding_proximal_adagrad_parameters_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_proximal_adagrad_parameters_fn get;

PythonFunctionContainer load_tpu_embedding_proximal_adagrad_parameters_grad_accum_debug_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_proximal_adagrad_parameters_grad_accum_debug_fn get;

PythonFunctionContainer load_tpu_embedding_rms_prop_parameters_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_rms_prop_parameters_fn get;

PythonFunctionContainer load_tpu_embedding_rms_prop_parameters_grad_accum_debug_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_rms_prop_parameters_grad_accum_debug_fn get;

PythonFunctionContainer load_tpu_embedding_stochastic_gradient_descent_parameters_eager_fallback_fn get;

PythonFunctionContainer load_tpu_embedding_stochastic_gradient_descent_parameters_fn get;

PythonFunctionContainer outfeed_dequeue_eager_fallback_fn get;

PythonFunctionContainer outfeed_dequeue_fn get;

PythonFunctionContainer outfeed_dequeue_tuple_eager_fallback_fn get;

PythonFunctionContainer outfeed_dequeue_tuple_fn get;

PythonFunctionContainer outfeed_enqueue_eager_fallback_fn get;

PythonFunctionContainer outfeed_enqueue_fn get;

PythonFunctionContainer outfeed_enqueue_tuple_eager_fallback_fn get;

PythonFunctionContainer outfeed_enqueue_tuple_fn get;

PythonFunctionContainer prelinearize_eager_fallback_fn get;

PythonFunctionContainer prelinearize_fn get;

PythonFunctionContainer prelinearize_tuple_eager_fallback_fn get;

PythonFunctionContainer prelinearize_tuple_fn get;

PythonFunctionContainer recv_tpu_embedding_activations_eager_fallback_fn get;

PythonFunctionContainer recv_tpu_embedding_activations_fn get;

PythonFunctionContainer retrieve_tpu_embedding_adadelta_parameters_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_adadelta_parameters_fn get;

PythonFunctionContainer retrieve_tpu_embedding_adadelta_parameters_grad_accum_debug_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_adadelta_parameters_grad_accum_debug_fn get;

PythonFunctionContainer retrieve_tpu_embedding_adagrad_parameters_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_adagrad_parameters_fn get;

PythonFunctionContainer retrieve_tpu_embedding_adagrad_parameters_grad_accum_debug_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_adagrad_parameters_grad_accum_debug_fn get;

PythonFunctionContainer retrieve_tpu_embedding_adam_parameters_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_adam_parameters_fn get;

PythonFunctionContainer retrieve_tpu_embedding_adam_parameters_grad_accum_debug_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_adam_parameters_grad_accum_debug_fn get;

PythonFunctionContainer retrieve_tpu_embedding_centered_rms_prop_parameters_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_centered_rms_prop_parameters_fn get;

PythonFunctionContainer retrieve_tpu_embedding_ftrl_parameters_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_ftrl_parameters_fn get;

PythonFunctionContainer retrieve_tpu_embedding_ftrl_parameters_grad_accum_debug_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_ftrl_parameters_grad_accum_debug_fn get;

PythonFunctionContainer retrieve_tpu_embedding_mdl_adagrad_light_parameters_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_mdl_adagrad_light_parameters_fn get;

PythonFunctionContainer retrieve_tpu_embedding_momentum_parameters_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_momentum_parameters_fn get;

PythonFunctionContainer retrieve_tpu_embedding_momentum_parameters_grad_accum_debug_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_momentum_parameters_grad_accum_debug_fn get;

PythonFunctionContainer retrieve_tpu_embedding_proximal_adagrad_parameters_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_proximal_adagrad_parameters_fn get;

PythonFunctionContainer retrieve_tpu_embedding_proximal_adagrad_parameters_grad_accum_debug_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_proximal_adagrad_parameters_grad_accum_debug_fn get;

PythonFunctionContainer retrieve_tpu_embedding_rms_prop_parameters_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_rms_prop_parameters_fn get;

PythonFunctionContainer retrieve_tpu_embedding_rms_prop_parameters_grad_accum_debug_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_rms_prop_parameters_grad_accum_debug_fn get;

PythonFunctionContainer retrieve_tpu_embedding_stochastic_gradient_descent_parameters_eager_fallback_fn get;

PythonFunctionContainer retrieve_tpu_embedding_stochastic_gradient_descent_parameters_fn get;

PythonFunctionContainer send_tpu_embedding_gradients_eager_fallback_fn get;

PythonFunctionContainer send_tpu_embedding_gradients_fn get;

PythonFunctionContainer shutdown_distributed_tpu_eager_fallback_fn get;

PythonFunctionContainer shutdown_distributed_tpu_fn get;

PythonFunctionContainer tpu_compilation_result_eager_fallback_fn get;

PythonFunctionContainer tpu_compilation_result_fn get;

PythonFunctionContainer tpu_embedding_activations_eager_fallback_fn get;

PythonFunctionContainer tpu_embedding_activations_fn get;

PythonFunctionContainer tpu_ordinal_selector_eager_fallback_fn get;

PythonFunctionContainer tpu_ordinal_selector_fn get;

PythonFunctionContainer tpu_partitioned_call_eager_fallback_fn get;

PythonFunctionContainer tpu_partitioned_call_fn get;

PythonFunctionContainer tpu_replicate_metadata_eager_fallback_fn get;

PythonFunctionContainer tpu_replicate_metadata_fn get;

PythonFunctionContainer tpu_replicated_input_eager_fallback_fn get;

PythonFunctionContainer tpu_replicated_input_fn get;

PythonFunctionContainer tpu_replicated_output_eager_fallback_fn get;

PythonFunctionContainer tpu_replicated_output_fn get;

PythonFunctionContainer worker_heartbeat_eager_fallback_fn get;

PythonFunctionContainer worker_heartbeat_fn get;