| /external/tensorflow/tensorflow/python/ops/ |
| quantized_conv_ops_test.py | 34 def _VerifyValues(self, tensor_in_sizes, filter_in_sizes, stride, padding,
41 filter_in_sizes: Filter tensor dimensions in
51 for s in filter_in_sizes:
60 x2 = x2.astype(np.uint8).reshape(filter_in_sizes)
65 t2 = constant_op.constant(x2, shape=filter_in_sizes, dtype=dtypes.quint8)
134 filter_in_sizes=[1, 1, 3, 3],
164 filter_in_sizes=[2, 2, 3, 3],
178 filter_in_sizes=[1, 2, 3, 3],
188 filter_in_sizes=[2, 2, 3, 3],
198 filter_in_sizes=[2, 2, 3, 3]
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| conv_ops_3d_test.py | 61 def _SetupValuesForDevice(self, tensor_in_sizes, filter_in_sizes, stride,
67 for s in filter_in_sizes:
77 t2 = constant_op.constant(x2, shape=filter_in_sizes, dtype=dtype)
94 def _VerifyValues(self, tensor_in_sizes, filter_in_sizes, stride, padding,
101 filter_in_sizes,
130 filter_in_sizes=[1, 1, 1, 3, 3],
136 filter_in_sizes=[1, 1, 1, 3, 3],
142 filter_in_sizes=[1, 1, 1, 3, 3],
158 filter_in_sizes=[2, 2, 2, 3, 3], # z, y, x, fin, fout
179 filter_in_sizes=[1, 2, 3, 1, 1]
[all...] |
| neon_depthwise_conv_op_test.py | 91 filter_in_sizes,
101 filter_in_sizes: Filter tensor dimensions in
112 for s in filter_in_sizes:
121 t2 = constant_op.constant(x2, shape=filter_in_sizes)
148 print("depthwise conv_2d: ", tensor_in_sizes, "*", filter_in_sizes,
183 def _VerifyHandValues(self, tensor_in_sizes, filter_in_sizes, stride, padding,
190 filter_in_sizes: Filter tensor dimensions in
201 for s in filter_in_sizes:
211 t2 = constant_op.constant(x2, shape=filter_in_sizes)
271 filter_in_sizes=[2, 2, 2, 2]
[all...] |
| conv_ops_test.py | 168 def _SetupValuesForDevice(self, tensor_in_sizes, filter_in_sizes, dilations,
175 filter_in_sizes: Filter tensor dimensions in
190 for s in filter_in_sizes:
199 t2 = constant_op.constant(x2, shape=filter_in_sizes, dtype=dtype)
218 def _CompareFwdValues(self, tensor_in_sizes, filter_in_sizes, conv_strides,
225 filter_in_sizes: Filter tensor dimensions in
231 x2 = np.random.rand(*filter_in_sizes).astype(np.float32)
236 t2 = constant_op.constant(x2, shape=filter_in_sizes)
254 def _ComputeReferenceDilatedConv(self, tensor_in_sizes, filter_in_sizes,
261 for s in filter_in_sizes
[all...] |
| depthwise_conv_op_test.py | 96 filter_in_sizes,
107 filter_in_sizes: Filter tensor dimensions in
119 for s in filter_in_sizes:
135 t2 = constant_op.constant(x2, shape=filter_in_sizes, dtype=data_type)
198 def _VerifyHandValues(self, tensor_in_sizes, filter_in_sizes, stride, padding,
205 filter_in_sizes: Filter tensor dimensions in
216 for s in filter_in_sizes:
225 t2 = constant_op.constant(x2, shape=filter_in_sizes)
285 filter_in_sizes=[2, 2, 2, 2],
293 filter_in_sizes=[2, 2, 2, 2]
[all...] |
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| fused_conv2d_bias_activation_op_test.py | 165 def _SetupValuesForDevice(self, tensor_in_sizes, filter_in_sizes, bias,
173 filter_in_sizes: Filter tensor dimensions in
187 filter_size = np.prod(filter_in_sizes)
188 bias_size = filter_in_sizes[-1] # equals to output depth
198 t2 = constant_op.constant(x2, shape=filter_in_sizes, dtype=dtype)
227 def _CompareFwdValues(self, tensor_in_sizes, filter_in_sizes, conv_strides,
234 filter_in_sizes: Filter tensor dimensions in
240 x2 = np.random.rand(*filter_in_sizes).astype(np.float32)
241 x3 = np.random.rand(*[filter_in_sizes[-1]]).astype(np.float32)
246 t2 = constant_op.constant(x2, shape=filter_in_sizes)
[all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| depthwise_conv_op_test.py | 126 filter_in_sizes,
136 filter_in_sizes: Filter tensor dimensions in
147 for s in filter_in_sizes:
153 dtype=data_type).reshape(filter_in_sizes)
162 t2 = array_ops.placeholder(shape=filter_in_sizes, dtype=data_type)
225 def _VerifyHandValues(self, tensor_in_sizes, filter_in_sizes, stride, padding,
232 filter_in_sizes: Filter tensor dimensions in
242 for s in filter_in_sizes:
249 dtype=np.float32).reshape(filter_in_sizes)
252 t2 = array_ops.placeholder(shape=filter_in_sizes, dtype=np.float32
[all...] |
