| /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
83 numpy_flat = nest.flatten(numpy_input)
90 vars_nested = nest.pack_sequence_as(numpy_input, vars_flat)
|
| D | distribute_lib.py | 1888 def experimental_make_numpy_dataset(self, numpy_input, session=None): argument
1917 numpy_input, session=session)
2632 def experimental_make_numpy_dataset(self, numpy_input, session=None): argument
2650 return self._experimental_make_numpy_dataset(numpy_input, session=session)
2652 def _experimental_make_numpy_dataset(self, numpy_input, session): argument
3443 def _experimental_make_numpy_dataset(self, numpy_input, session): argument
3444 numpy_flat = nest.flatten(numpy_input)
3452 vars_nested = nest.pack_sequence_as(numpy_input, vars_flat)
|
| D | one_device_strategy.py | 304 def _experimental_make_numpy_dataset(self, numpy_input, session): argument
306 numpy_input, numpy_dataset.SingleDevice(self._input_device), session)
|
| D | parameter_server_strategy.py | 376 def _experimental_make_numpy_dataset(self, numpy_input, session): argument
378 numpy_input, self._input_host_device, session)
|
| D | mirrored_strategy.py | 573 def _experimental_make_numpy_dataset(self, numpy_input, session): argument
575 numpy_input, self._host_input_device, session)
|
| D | tpu_strategy.py | 906 def _experimental_make_numpy_dataset(self, numpy_input, session): argument
908 numpy_input, numpy_dataset.SingleDevice(self._host_device),
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | data_adapter_test.py | 75 self.numpy_input = np.zeros((50, 10))
79 self.arraylike_input = DummyArrayLike(self.numpy_input)
82 (self.numpy_input, self.numpy_target)).shuffle(50).batch(
119 self.assertTrue(self.adapter_cls.can_handle(self.numpy_input))
121 self.adapter_cls.can_handle(self.numpy_input, self.numpy_target))
131 self.numpy_input, self.numpy_target, batch_size=5)
137 self.numpy_input, self.numpy_target, batch_size=5)
142 self.numpy_input, self.numpy_target, batch_size=4)
150 self.numpy_input, self.numpy_target, batch_size=5, epochs=num_epochs)
152 num_batches_per_epoch = self.numpy_input.shape[0] // 5
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 56 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|
| D | tensorflow.distribute.-strategy.pbtxt | 48 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 49 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 49 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|
| D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 49 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 49 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 53 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|
| D | tensorflow.distribute.experimental.-central-storage-strategy.pbtxt | 49 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
|