| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | conv_ops_test.py | 182 def _SetupValuesForDevice(self, tensor_in_sizes, filter_in_sizes, dilations, argument
200 x1 = self._CreateNumpyTensor(tensor_in_sizes)
204 t1 = constant_op.constant(x1, shape=tensor_in_sizes, dtype=dtype)
229 def _CompareFwdValues(self, tensor_in_sizes, filter_in_sizes, conv_strides, argument
241 x1 = np.random.rand(*tensor_in_sizes).astype(np.float32)
246 t1 = constant_op.constant(x1, shape=tensor_in_sizes)
265 def _ComputeReferenceDilatedConv(self, tensor_in_sizes, filter_in_sizes, argument
268 x1 = self._CreateNumpyTensor(tensor_in_sizes)
271 t1 = constant_op.constant(x1, shape=tensor_in_sizes)
303 def _VerifyDilatedConvValues(self, tensor_in_sizes, filter_in_sizes, strides, argument
[all …]
|
| H A D | conv_ops_3d_test.py | 68 def _SetupValuesForDevice(self, tensor_in_sizes, filter_in_sizes, stride, argument
70 total_size_tensor = np.prod(tensor_in_sizes)
79 t1 = constant_op.constant(x1, shape=tensor_in_sizes, dtype=dtype)
97 def _VerifyValues(self, tensor_in_sizes, filter_in_sizes, stride, padding, argument
103 tensor_in_sizes,
123 def _ComputeReferenceDilatedConv(self, tensor_in_sizes, filter_in_sizes, argument
126 total_size_tensor = np.prod(tensor_in_sizes)
134 t1 = constant_op.constant(x1, shape=tensor_in_sizes)
166 def _VerifyDilatedConvValues(self, tensor_in_sizes, filter_in_sizes, stride, argument
178 tensor_in_sizes, filter_in_sizes, stride, dilations, padding,
[all …]
|
| H A D | depthwise_conv_op_base.py | 300 tensor_in_sizes, argument
328 for s in tensor_in_sizes:
334 x1 = np.array(x1).reshape(tensor_in_sizes)
354 t1 = constant_op.constant(x1, shape=tensor_in_sizes, dtype=data_type)
530 def _VerifyHandValues(self, tensor_in_sizes, filter_in_sizes, stride, padding, argument
546 for s in tensor_in_sizes:
555 t1 = constant_op.constant(x1, shape=tensor_in_sizes)
556 t1.set_shape(tensor_in_sizes)
616 tensor_in_sizes=[1, 2, 3, 2],
624 tensor_in_sizes=[1, 2, 3, 2],
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/image_ops/ |
| H A D | attention_ops_test.py | 30 def _VerifyValues(self, tensor_in_sizes, glimpse_sizes, offsets, argument
45 rows = tensor_in_sizes[0]
46 cols = tensor_in_sizes[1]
111 tensor_in_sizes=[41, 61],
127 tensor_in_sizes=[41, 61],
135 tensor_in_sizes=[41, 61],
143 tensor_in_sizes=[41, 61],
151 tensor_in_sizes=[41, 61],
159 tensor_in_sizes=[41, 61],
167 tensor_in_sizes=[41, 61],
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | quantized_conv_ops_test.py | 32 def _VerifyValues(self, tensor_in_sizes, filter_in_sizes, stride, padding, argument
47 for s in tensor_in_sizes:
54 x1 = x1.astype(np.uint8).reshape(tensor_in_sizes)
62 t1 = constant_op.constant(x1, shape=tensor_in_sizes, dtype=dtypes.quint8)
131 tensor_in_sizes=[1, 2, 3, 3],
161 tensor_in_sizes=[1, 2, 3, 3],
175 tensor_in_sizes=[1, 2, 3, 3],
185 tensor_in_sizes=[1, 2, 3, 3],
195 tensor_in_sizes=[1, 2, 3, 3],
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | depthwise_conv_op_test.py | 148 tensor_in_sizes, argument
168 for s in tensor_in_sizes:
174 dtype=data_type).reshape(tensor_in_sizes)
184 t1 = array_ops.placeholder(shape=tensor_in_sizes, dtype=data_type)
252 def _VerifyHandValues(self, tensor_in_sizes, filter_in_sizes, stride, padding, argument
267 for s in tensor_in_sizes:
274 dtype=np.float32).reshape(tensor_in_sizes)
278 t1 = array_ops.placeholder(shape=tensor_in_sizes, dtype=np.float32)
339 tensor_in_sizes=[1, 2, 3, 2],
349 tensor_in_sizes, argument
[all …]
|