Public static methods
object initialize_system_for_tpu_embedding(object embedding_config, object job)
object initialize_system_for_tpu_embedding_dyn(object embedding_config, object job)
void prune_unconnected_ops_from_xla(Graph prune_graph)
object prune_unconnected_ops_from_xla_dyn(object prune_graph)
object rewrite_for_inference(object computation, object inputs, object infeed_queue, object device_assignment, string name)
object rewrite_for_inference_dyn(object computation, object inputs, object infeed_queue, object device_assignment, object name)
IList<object> split_compile_and_replicate(object computation, IEnumerable<object> inputs, object infeed_queue, DeviceAssignment device_assignment, string name, bool use_tpu, int maximum_shapes)
IList<object> split_compile_and_replicate(object computation, IEnumerable<object> inputs, object infeed_queue, DeviceAssignment device_assignment, string name, bool use_tpu, IEnumerable<TensorShape> maximum_shapes)
IList<object> split_compile_and_replicate(object computation, IEnumerable<object> inputs, object infeed_queue, DeviceAssignment device_assignment, string name, bool use_tpu, TensorShape maximum_shapes)
IList<object> split_compile_and_replicate(object computation, IEnumerable<object> inputs, object infeed_queue, DeviceAssignment device_assignment, string name, bool use_tpu, PythonClassContainer maximum_shapes)
IList<object> split_compile_and_replicate(object computation, IEnumerable<object> inputs, object infeed_queue, DeviceAssignment device_assignment, string name, bool use_tpu, Dimension maximum_shapes)
object split_compile_and_replicate_dyn(object computation, object inputs, object infeed_queue, object device_assignment, object name, ImplicitContainer<T> use_tpu, object maximum_shapes)
ValueTuple<object, IList<object>> split_compile_and_shard(object computation, IEnumerable<object> inputs, int num_shards, object input_shard_axes, bool outputs_from_all_shards, object output_shard_axes, object infeed_queue, object device_assignment, string name)
object split_compile_and_shard_dyn(object computation, object inputs, ImplicitContainer<T> num_shards, object input_shard_axes, ImplicitContainer<T> outputs_from_all_shards, object output_shard_axes, object infeed_queue, object device_assignment, object name)
bool under_tpu_inference_context()
object under_tpu_inference_context_dyn()
void validate_inference_rewrite_for_variables(object graph)
object validate_inference_rewrite_for_variables_dyn(object graph)
Public properties