LostTech.TensorFlow : API Documentation

Type tpu_function

Namespace tensorflow.python.tpu.tpu_function

Methods

Properties

Public static methods

TpuContext get_tpu_context()

object get_tpu_context_dyn()

object on_device_training_loop(object func)

object on_device_training_loop_dyn(object func)

IContextManager<T> tpu_shard_context(int number_of_shards)

object tpu_shard_context_dyn(object number_of_shards)

Public properties

PythonFunctionContainer get_tpu_context_fn get;

PythonFunctionContainer on_device_training_loop_fn get;

PythonFunctionContainer tpu_shard_context_fn get;

PythonFunctionContainer TpuContext_fn get;