Home
last modified time | relevance | path

Searched refs:numpy_input (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dnumpy_dataset.py28 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 Ddistribute_lib.py1945 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 Done_device_strategy.py302 def _experimental_make_numpy_dataset(self, numpy_input, session): argument
304 numpy_input, numpy_dataset.SingleDevice(self._input_device), session)
H A Dparameter_server_strategy.py375 def _experimental_make_numpy_dataset(self, numpy_input, session): argument
377 numpy_input, self._input_host_device, session)
H A Dmirrored_strategy.py579 def _experimental_make_numpy_dataset(self, numpy_input, session): argument
581 numpy_input, self._host_input_device, session)
H A Dtpu_strategy.py962 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 Dtensorflow.distribute.-strategy-extended.pbtxt56 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
H A Dtensorflow.distribute.-strategy.pbtxt48 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
H A Dtensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt49 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
H A Dtensorflow.distribute.-one-device-strategy.pbtxt49 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
H A Dtensorflow.distribute.experimental.-t-p-u-strategy.pbtxt53 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
H A Dtensorflow.distribute.experimental.-central-storage-strategy.pbtxt49 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
H A Dtensorflow.distribute.experimental.-parameter-server-strategy.pbtxt49 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…
H A Dtensorflow.distribute.-mirrored-strategy.pbtxt49 …argspec: "args=[\'self\', \'numpy_input\', \'session\'], varargs=None, keywords=None, defaults=[\'…