| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_conv_ops_test.cc | 60 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 …]
|
| D | conv_ops_test.cc | 119 const int image_batch_count = 1; in HandwrittenConv() local
124 Tensor image(dtype, {image_batch_count, image_height, image_width, depth}); in HandwrittenConv()
173 Tensor expected(dtype, TensorShape({image_batch_count, expected_height, in HandwrittenConv()
434 const int image_batch_count = 1; in HandwrittenConv() local
440 {image_batch_count, image_height, image_width, depth}); in HandwrittenConv()
477 Tensor expected(DT_FLOAT, TensorShape({image_batch_count, expected_height, in HandwrittenConv()
499 const int image_batch_count = 1; in AnisotropicStrides() local
501 {image_batch_count, image_height, image_width, depth}); in AnisotropicStrides()
521 Tensor expected(DT_FLOAT, TensorShape({image_batch_count, expected_height, in AnisotropicStrides()
781 int image_batch_count, int filter_size, in VerifyBiasAddTensorsNear() argument
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_quantized_conv_ops_test.cc | 102 const int image_batch_count = 1; in RunQuantizedDepthwiseConv2DOp() local
108 TensorShape({image_batch_count, image_height, image_width, depth}), in RunQuantizedDepthwiseConv2DOp()
161 Tensor expected(DT_QINT32, TensorShape({image_batch_count, image_height, in RunQuantizedDepthwiseConv2DOp()
191 const int image_batch_count = 1; in TEST_F() local
202 {image_batch_count, image_height, image_width, depth}); in TEST_F()
267 DT_FLOAT, TensorShape({image_batch_count, expected_height, expected_width, in TEST_F()
293 const int image_batch_count = 1; in TEST_F() local
328 {image_batch_count, image_height, image_width, depth}); in TEST_F()
372 DT_FLOAT, TensorShape({image_batch_count, expected_height, expected_width, in TEST_F()
401 const int image_batch_count = 1; in TEST_F() local
[all …]
|
| D | mkl_quantized_conv_ops_perchannel_test.cc | 101 const int image_batch_count = 1; in TEST_F() local
115 DT_FLOAT, {image_batch_count, image_height, image_width, image_channel}); in TEST_F()
208 DT_FLOAT, TensorShape({image_batch_count, expected_height, expected_width, in TEST_F()
|
| D | mkl_fused_ops_test.cc | 117 int image_height, int image_batch_count, in VerifyBiasAddTensorsClose() argument
123 Tensor image(dtype, {image_batch_count, image_height, image_width, depth}); in VerifyBiasAddTensorsClose()
146 int depth, int image_width, int image_height, int image_batch_count, in VerifyFusedTensorsClose() argument
153 Tensor image(dtype, {image_batch_count, image_height, image_width, depth}); in VerifyFusedTensorsClose()
348 int image_batch_count = kImageBatchCount) { in VerifyFusedConv2D() argument
372 depth, image_width, image_height, image_batch_count, filter_size, in VerifyFusedConv2D()
671 int image_batch_count = kImageBatchCount) { in VerifyFusedDepthwiseConv2D() argument
690 depth, image_width, image_height, image_batch_count, filter_size, in VerifyFusedDepthwiseConv2D()
791 const int image_batch_count = 1; in Run() local
797 Tensor(dtype, {image_batch_count, image_height, image_width, depth}); in Run()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | conv_test.cc | 512 const int image_batch_count = 1; in TEST_P() local
521 {image_batch_count, image_height, image_width, depth}}, in TEST_P()
587 const int image_batch_count = 1; in TEST_P() local
601 {image_batch_count, image_height, image_width, depth}}, in TEST_P()
657 const int image_batch_count = 1; in TEST_P() local
666 {image_batch_count, image_height, image_width, depth}}, in TEST_P()
711 const int image_batch_count = 1; in TEST_P() local
720 {image_batch_count, image_height, image_width, depth}}, in TEST_P()
767 const int image_batch_count = 1; in TEST_P() local
776 {image_batch_count, image_height, image_width, depth}}, in TEST_P()
[all …]
|
| D | depthwise_conv_test.cc | 372 const int image_batch_count = 1; in SimpleDilatedTestPaddingValid() local
379 {image_batch_count, image_height, image_width, depth}}, in SimpleDilatedTestPaddingValid()
434 const int image_batch_count = 1; in SimpleDilatedTestPaddingSame() local
441 {image_batch_count, image_height, image_width, depth}}, in SimpleDilatedTestPaddingSame()
803 const int image_batch_count = 1; in TEST_P() local
810 {image_batch_count, image_height, image_width, depth}, in TEST_P()
862 const int image_batch_count = 1; in TEST_P() local
869 {image_batch_count, image_height, image_width, depth}, in TEST_P()
903 const int image_batch_count = 3; in TEST_P() local
910 {image_batch_count, image_height, image_width, depth}, in TEST_P()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | conv_test.cc | 343 const int image_batch_count = 1; in TEST() local
350 {image_batch_count, image_height, image_width, depth}, in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 808 const int image_batch_count = 1; in TEST() local
818 {image_batch_count, image_height, image_width, depth}}, in TEST()
889 const int image_batch_count = 1; in TEST() local
898 {image_batch_count, image_height, image_width, depth}, in TEST()
|