Searched defs:image_batch_count (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_quantized_conv_ops_test.cc | 63 const int image_batch_count = 1; in RunQuantizedDepthwiseConv2DOp() local
134 const int image_batch_count = 1; in TEST_F() local
221 const int image_batch_count = 1; in TEST_F() local
306 const int image_batch_count = 1; in TEST_F() local
365 const int image_batch_count = 1; in TEST_F() local
407 const int image_batch_count = 1; in TEST_F() local
447 const int image_batch_count = 3; in TEST_F() local
|
| H A D | mkl_fused_ops_test.cc | 119 int image_height, int image_batch_count, in VerifyBiasAddTensorsClose()
148 int depth, int image_width, int image_height, int image_batch_count, in VerifyFusedTensorsClose()
328 int image_batch_count = kImageBatchCount) { in VerifyFusedConv2D()
619 int image_batch_count = kImageBatchCount) { in VerifyFusedDepthwiseConv2D()
746 const int image_batch_count = 1; in Run() local
870 const int image_batch_count = 1; in TEST_F() local
1228 const int image_batch_count = 1; in TEST_F() local
1285 int image_batch_count = kImageBatchCount) { in VerifyPadAndConv2DWithBias()
1310 int image_batch_count = kImageBatchCount) { in VerifyPadAndConv2DWithBiasRelu()
|
| H A D | mkl_quantized_conv_ops_perchannel_test.cc | 63 const int image_batch_count = 1; in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantized_conv_ops_test.cc | 60 const int image_batch_count = 1; in TEST_F() local
152 const int image_batch_count = 1; in TEST_F() local
195 const int image_batch_count = 1; in TEST_F() local
236 const int image_batch_count = 3; in TEST_F() local
280 const int image_batch_count = 1; in TEST_F() local
|
| H A D | conv_ops_test.cc | 62 const int image_batch_count = 1; in HandwrittenConv() local
377 const int image_batch_count = 1; in HandwrittenConv() local
442 const int image_batch_count = 1; in AnisotropicStrides() local
724 int image_batch_count, int filter_size, in VerifyBiasAddTensorsNear()
764 int image_height, int image_batch_count, in VerifyFusedBatchNormTensorsNear()
819 int image_batch_count = kImageBatchCount) {
848 int image_batch_count = kImageBatchCount) {
882 int image_batch_count = kImageBatchCount) {
915 int image_batch_count = kImageBatchCount) {
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | conv_test.cc | 539 const int image_batch_count = 1; in TEST_P() local
614 const int image_batch_count = 1; in TEST_P() local
684 const int image_batch_count = 1; in TEST_P() local
738 const int image_batch_count = 1; in TEST_P() local
794 const int image_batch_count = 1; in TEST_P() local
836 const int image_batch_count = 1; in TEST_P() local
1084 const int image_batch_count = 1; in TEST_P() local
|
| H A D | depthwise_conv_test.cc | 372 const int image_batch_count = 1; in SimpleDilatedTestPaddingValid() local
434 const int image_batch_count = 1; in SimpleDilatedTestPaddingSame() local
803 const int image_batch_count = 1; in TEST_P() local
862 const int image_batch_count = 1; in TEST_P() local
903 const int image_batch_count = 3; in TEST_P() local
993 const int image_batch_count = 6; in TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| H A D | conv_test.cc | 357 const int image_batch_count = 1; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate_test.cc | 885 const int image_batch_count = 1; in TEST() local
966 const int image_batch_count = 1; in TEST() local
|