Searched defs:input_batches (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | eigen_benchmark_cpu_test.cc | 32 int input_batches, int input_height, int input_width, in SpatialConvolution()
59 int input_batches, int input_height, in SpatialConvolutionBackwardInput()
86 int input_batches, int input_height, in SpatialConvolutionBackwardKernel()
249 int input_batches, int input_height, int input_width, in CuboidConvolution()
277 int input_batches, int input_height, in CuboidConvolutionBackwardInput()
305 int input_batches, int input_height, in CuboidConvolutionBackwardKernel()
|
| H A D | conv_ops_using_gemm.cc | 91 int input_batches, int input_height, int input_width, in operator ()()
214 int input_batches, int input_height, int input_width, in operator ()()
|
| H A D | quantized_conv_ops.cc | 58 int input_batches, int input_height, int input_width, in operator ()()
204 int input_batches, int input_height, int input_width, in operator ()()
|
| H A D | conv_ops_fused_image_transform.cc | 261 int input_batches, int resized_height, int resized_width, in operator ()()
|
| H A D | eigen_spatial_convolutions_test.cc | 1381 int input_batches, int input_cols, int input_rows, in PackRhsHelper()
|
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_image_patch.cpp | 188 int input_batches = 1; in test_patch_padding_valid() local
264 int input_batches = 2; in test_patch_padding_valid_same_value() local
337 int input_batches = 1; in test_patch_padding_same() local
414 int input_batches = 1; in test_patch_padding_same_negative_padding_clip_to_zero() local
|
| H A D | cxx11_tensor_image_patch_sycl.cpp | 263 IndexType input_batches = 1; in test_patch_padding_valid_sycl() local
369 IndexType input_batches = 2; in test_patch_padding_valid_same_value_sycl() local
466 IndexType input_batches = 1; in test_patch_padding_same_sycl() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/ |
| H A D | CLInstanceNormalizationLayerKernel.cpp | 106 const unsigned int input_batches = input->info()->dimension(batches_idx); in configure() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_conv_ops_test.cc | 57 int input_batches; member
|