| /external/tensorflow/tensorflow/python/distribute/ |
| D | numpy_dataset.py | 32 def init_var_from_numpy(input_var, numpy_input, session): argument
36 input_var.assign(numpy_input)
57 np.prod(numpy_input.shape[1:]) * input_var.dtype.size)
66 limit = numpy_input.shape[0]
72 slice_placeholder: numpy_input[start:end]})
76 def one_host_numpy_dataset(numpy_input, colocate_with, session): argument
82 numpy_flat = nest.flatten(numpy_input)
89 vars_nested = nest.pack_sequence_as(numpy_input, vars_flat)
|
| D | distribute_lib.py | 407 self, numpy_input, batch_size, num_epochs=1, shuffle=1024, session=None): argument
429 numpy_input, session=session)
991 def experimental_make_numpy_dataset(self, numpy_input, session=None): argument
1009 return self._experimental_make_numpy_dataset(numpy_input, session=session)
1011 def _experimental_make_numpy_dataset(self, numpy_input, session): argument
1581 def _experimental_make_numpy_dataset(self, numpy_input, session): argument
1582 numpy_flat = nest.flatten(numpy_input)
1590 vars_nested = nest.pack_sequence_as(numpy_input, vars_flat)
|
| D | one_device_strategy.py | 89 def _experimental_make_numpy_dataset(self, numpy_input, session): argument
91 numpy_input, numpy_dataset.SingleDevice(self._input_device), session)
|
| D | parameter_server_strategy.py | 279 def _experimental_make_numpy_dataset(self, numpy_input, session): argument
281 numpy_input, self._input_host_device, session)
|
| D | tpu_strategy.py | 276 def _experimental_make_numpy_dataset(self, numpy_input, session): argument
278 numpy_input, numpy_dataset.SingleDevice(self._host_device),
|
| D | mirrored_strategy.py | 587 def _experimental_make_numpy_dataset(self, numpy_input, session): argument
589 numpy_input, self._host_input_device, session)
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | parameter_server_strategy.py | 120 self, numpy_input, batch_size, num_epochs=1, shuffle=1024, session=None): argument
146 numpy_input, batch_size, num_epochs, shuffle, session)
|
| D | mirrored_strategy.py | 133 self, numpy_input, batch_size, num_epochs=1, shuffle=1024, session=None): argument
158 numpy_input, batch_size, num_epochs, shuffle, session)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-strategy.pbtxt | 31 …argspec: "args=[\'self\', \'numpy_input\', \'batch_size\', \'num_epochs\', \'shuffle\', \'session\…
|
| D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 32 …argspec: "args=[\'self\', \'numpy_input\', \'batch_size\', \'num_epochs\', \'shuffle\', \'session\…
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 32 …argspec: "args=[\'self\', \'numpy_input\', \'batch_size\', \'num_epochs\', \'shuffle\', \'session\…
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 32 …argspec: "args=[\'self\', \'numpy_input\', \'batch_size\', \'num_epochs\', \'shuffle\', \'session\…
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 36 …argspec: "args=[\'self\', \'numpy_input\', \'batch_size\', \'num_epochs\', \'shuffle\', \'session\…
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 32 …argspec: "args=[\'self\', \'numpy_input\', \'batch_size\', \'num_epochs\', \'shuffle\', \'session\…
|
| D | tensorflow.distribute.-strategy-extended.pbtxt | 55 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy.pbtxt | 31 …argspec: "args=[\'self\', \'numpy_input\', \'batch_size\', \'num_epochs\', \'shuffle\', \'session\…
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 36 …argspec: "args=[\'self\', \'numpy_input\', \'batch_size\', \'num_epochs\', \'shuffle\', \'session\…
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 32 …argspec: "args=[\'self\', \'numpy_input\', \'batch_size\', \'num_epochs\', \'shuffle\', \'session\…
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 32 …argspec: "args=[\'self\', \'numpy_input\', \'batch_size\', \'num_epochs\', \'shuffle\', \'session\…
|
| D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 32 …argspec: "args=[\'self\', \'numpy_input\', \'batch_size\', \'num_epochs\', \'shuffle\', \'session\…
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 32 …argspec: "args=[\'self\', \'numpy_input\', \'batch_size\', \'num_epochs\', \'shuffle\', \'session\…
|
| D | tensorflow.distribute.-strategy-extended.pbtxt | 55 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|