| /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=[\'…
|