LostTech.TensorFlow : API Documentation

Type tpu

Namespace tensorflow.python.tpu.tpu

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

PythonFunctionContainer _TPUInferenceContext_fn get;

PythonFunctionContainer initialize_system_for_tpu_embedding_fn get;

PythonFunctionContainer prune_unconnected_ops_from_xla_fn get;

PythonFunctionContainer rewrite_for_inference_fn get;

PythonFunctionContainer split_compile_and_replicate_fn get;

PythonFunctionContainer split_compile_and_shard_fn get;

PythonFunctionContainer TPUReplicateContext_fn get;

PythonFunctionContainer under_tpu_inference_context_fn get;

PythonFunctionContainer validate_inference_rewrite_for_variables_fn get;