Home
last modified time | relevance | path

Searched refs:image_batch_count (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dquantized_conv_ops_test.cc60 const int image_batch_count = 1; in TEST_F() local
70 {image_batch_count, image_height, image_width, depth}); in TEST_F()
122 DT_FLOAT, TensorShape({image_batch_count, expected_height, expected_width, in TEST_F()
152 const int image_batch_count = 1; in TEST_F() local
154 TensorShape({image_batch_count, image_height, image_width, depth}), in TEST_F()
169 Tensor expected(DT_QINT32, TensorShape({image_batch_count, expected_height, in TEST_F()
195 const int image_batch_count = 1; in TEST_F() local
197 TensorShape({image_batch_count, image_height, image_width, depth}), in TEST_F()
212 Tensor expected(DT_QINT32, TensorShape({image_batch_count, expected_height, in TEST_F()
236 const int image_batch_count = 3; in TEST_F() local
[all …]
Dmkl_quantized_conv_ops_test.cc109 const int image_batch_count = 1; in TEST_F() local
120 {image_batch_count, image_height, image_width, depth}); in TEST_F()
185 DT_FLOAT, TensorShape({image_batch_count, expected_height, expected_width, in TEST_F()
216 const int image_batch_count = 1; in TEST_F() local
218 TensorShape({image_batch_count, image_height, image_width, depth}), in TEST_F()
247 Tensor expected(DT_QINT32, TensorShape({image_batch_count, expected_height, in TEST_F()
298 const int image_batch_count = 1; in TEST_F() local
300 TensorShape({image_batch_count, image_height, image_width, depth}), in TEST_F()
329 Tensor expected(DT_QINT32, TensorShape({image_batch_count, expected_height, in TEST_F()
354 const int image_batch_count = 1; in TEST_F() local
[all …]
Dmkl_fused_ops_test.cc93 int image_height, int image_batch_count, in VerifyBiasAddTensorsClose() argument
99 Tensor image(dtype, {image_batch_count, image_height, image_width, depth}); in VerifyBiasAddTensorsClose()
220 int image_batch_count = kImageBatchCount) { in VerifyConv2DWithBias() argument
235 depth, image_width, image_height, image_batch_count, filter_size, in VerifyConv2DWithBias()
245 int image_batch_count = kImageBatchCount) { in VerifyConv2DWithBiasAndRelu() argument
260 depth, image_width, image_height, image_batch_count, filter_size, in VerifyConv2DWithBiasAndRelu()
353 const int image_batch_count = 1; in TEST_F() local
354 Tensor image(DT_FLOAT, {image_batch_count, image_height, image_width, depth}); in TEST_F()
381 const int image_batch_count = 1; in TEST_F() local
382 Tensor image(DT_FLOAT, {image_batch_count, depth, image_height, image_width}); in TEST_F()
[all …]
Dconv_ops_test.cc116 const int image_batch_count = 1; in HandwrittenConv() local
121 Tensor image(dtype, {image_batch_count, image_height, image_width, depth}); in HandwrittenConv()
170 Tensor expected(dtype, TensorShape({image_batch_count, expected_height, in HandwrittenConv()
431 const int image_batch_count = 1; in HandwrittenConv() local
437 {image_batch_count, image_height, image_width, depth}); in HandwrittenConv()
474 Tensor expected(DT_FLOAT, TensorShape({image_batch_count, expected_height, in HandwrittenConv()
496 const int image_batch_count = 1; in AnisotropicStrides() local
498 {image_batch_count, image_height, image_width, depth}); in AnisotropicStrides()
518 Tensor expected(DT_FLOAT, TensorShape({image_batch_count, expected_height, in AnisotropicStrides()
745 int image_batch_count, int filter_size, in VerifyBiasAddTensorsNear() argument
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Ddepthwise_conv_test.cc177 const int image_batch_count = 1; in TEST_P() local
184 {image_batch_count, image_height, image_width, depth}}, in TEST_P()
230 const int image_batch_count = 1; in TEST_P() local
237 {image_batch_count, image_height, image_width, depth}}, in TEST_P()
370 const int image_batch_count = 1; in TEST_P() local
377 {image_batch_count, image_height, image_width, depth}, in TEST_P()
429 const int image_batch_count = 1; in TEST_P() local
436 {image_batch_count, image_height, image_width, depth}, in TEST_P()
Dconv_test.cc319 const int image_batch_count = 1; in TEST_P() local
328 {image_batch_count, image_height, image_width, depth}}, in TEST_P()
385 const int image_batch_count = 1; in TEST_P() local
394 {image_batch_count, image_height, image_width, depth}}, in TEST_P()
439 const int image_batch_count = 1; in TEST_P() local
448 {image_batch_count, image_height, image_width, depth}}, in TEST_P()
495 const int image_batch_count = 1; in TEST_P() local
504 {image_batch_count, image_height, image_width, depth}}, in TEST_P()
537 const int image_batch_count = 1; in TEST_P() local
548 {image_batch_count, image_height, image_width, depth}}, in TEST_P()
[all …]