/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
H A D | numpy_dataset.py | 28 def init_var_from_numpy(input_var, numpy_input, session): argument 32 input_var.assign(numpy_input) 53 np.prod(numpy_input.shape[1:]) * input_var.dtype.size) 62 limit = numpy_input.shape[0] 68 slice_placeholder: numpy_input[start:end]}) 72 def one_host_numpy_dataset(numpy_input, colocate_with, session): argument 79 numpy_flat = nest.flatten(numpy_input) 86 vars_nested = nest.pack_sequence_as(numpy_input, vars_flat)
|
H A D | distribute_lib.py | 1945 def experimental_make_numpy_dataset(self, numpy_input, session=None): argument 1974 numpy_input, session=session) 2748 def experimental_make_numpy_dataset(self, numpy_input, session=None): argument 2766 return self._experimental_make_numpy_dataset(numpy_input, session=session) 2768 def _experimental_make_numpy_dataset(self, numpy_input, session): argument 3672 def _experimental_make_numpy_dataset(self, numpy_input, session): argument 3673 numpy_flat = nest.flatten(numpy_input) 3681 vars_nested = nest.pack_sequence_as(numpy_input, vars_flat)
|
H A D | one_device_strategy.py | 302 def _experimental_make_numpy_dataset(self, numpy_input, session): argument 304 numpy_input, numpy_dataset.SingleDevice(self._input_device), session)
|
H A D | parameter_server_strategy.py | 375 def _experimental_make_numpy_dataset(self, numpy_input, session): argument 377 numpy_input, self._input_host_device, session)
|
H A D | mirrored_strategy.py | 579 def _experimental_make_numpy_dataset(self, numpy_input, session): argument 581 numpy_input, self._host_input_device, session)
|
H A D | tpu_strategy.py | 962 def _experimental_make_numpy_dataset(self, numpy_input, session): argument 964 numpy_input, numpy_dataset.SingleDevice(self._host_device),
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
H A D | tensorflow.distribute.-strategy-extended.pbtxt | 56 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|
H A D | tensorflow.distribute.-strategy.pbtxt | 48 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|
H A D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 49 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|
H A D | tensorflow.distribute.-one-device-strategy.pbtxt | 49 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|
H A D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 53 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|
H A D | tensorflow.distribute.experimental.-central-storage-strategy.pbtxt | 49 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|
H A D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 49 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|
H A D | tensorflow.distribute.-mirrored-strategy.pbtxt | 49 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|