Home
last modified time | relevance | path

Searched refs:inp_array (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Dlrn_op_test.py107 inp_array = np.zeros(shape).astype("f")
110 params = {p: inp_array}
165 inp_array = np.random.rand(*shape).astype(np.float32)
167 inp_array = np.random.rand(*shape).astype(np.float16)
170 list(inp_array.ravel(order="C")), shape=shape, dtype=dtype)
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dcustom_training_loop_gradient_test.py30 def get_dataset_from_tensor_slices(inp_array): argument
31 dataset = dataset_ops.DatasetV2.from_tensor_slices(inp_array)
34 dataset = dataset_ops.Dataset.from_tensor_slices(inp_array)
H A Dcustom_training_loop_input_test.py43 def get_dataset_from_tensor_slices(inp_array): argument
44 dataset = dataset_ops.DatasetV2.from_tensor_slices(inp_array)
47 dataset = dataset_ops.Dataset.from_tensor_slices(inp_array)