| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_convolution_sycl.cpp | 45 Eigen::array<IndexType, 1> kernel_dims = {{4}}; in test_larg_expr1D() local
106 Eigen::array<IndexType, 2> kernel_dims = {{4,5}}; in test_larg_expr2D() local
167 Eigen::array<IndexType, 3> kernel_dims = {{4,5,3}}; in test_larg_expr3D() local
222 Eigen::array<IndexType, 1> kernel_dims = {{2}}; in test_evals() local
268 Eigen::array<IndexType, 2> kernel_dims = {{2, 2}}; in test_expr() local
317 Eigen::array<IndexType, 1> kernel_dims = {{3}}; in test_modes() local
406 Eigen::array<IndexType, 1> kernel_dims = {{3}}; in test_strides() local
|
| /aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/ |
| H A D | ClIm2ColKernel.cpp | 62 Status validate_arguments(const ITensorInfo *src, const ITensorInfo *dst, const Size2D &kernel_dims… in validate_arguments()
95 …lidate_and_configure_window(ITensorInfo *src, ITensorInfo *dst, const Size2D &kernel_dims, const P… in validate_and_configure_window()
148 Im2ColConfiguration configure_opencl_kernel(const ITensorInfo *src, const Size2D &kernel_dims, cons… in configure_opencl_kernel()
304 …leContext &compile_context, ITensorInfo *src, ITensorInfo *dst, const Size2D &kernel_dims, const P… in configure()
355 …ernel::validate(const ITensorInfo *src, const ITensorInfo *dst, const Size2D &kernel_dims, const P… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | Im2Col.cpp | 39 void im2col_nchw(const SimpleTensor<T> &src, SimpleTensor<T> &dst, const Size2D &kernel_dims, const… in im2col_nchw()
93 void im2col_nhwc(const SimpleTensor<T> &src, SimpleTensor<T> &dst, const Size2D &kernel_dims, const… in im2col_nhwc()
144 void im2col(const SimpleTensor<T> &src, SimpleTensor<T> &dst, const Size2D &kernel_dims, const PadS… in im2col()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuIm2ColKernel.cpp | 54 …_arguments(const ITensorInfo *input, const ITensorInfo *output, const Size2D &kernel_dims, const P… in validate_arguments()
334 void CpuIm2ColKernel::configure(const ITensorInfo *src, ITensorInfo *dst, const Size2D &kernel_dims… in configure()
425 …ernel::validate(const ITensorInfo *src, const ITensorInfo *dst, const Size2D &kernel_dims, const P… in validate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | runtime_conv_impl.h | 85 Eigen::DSizes<Eigen::Index, 3> kernel_dims; in EigenConv2DImpl() local
170 Eigen::DSizes<Eigen::Index, 3> kernel_dims; in EigenConv3DImpl() local
|
| H A D | ir_emitter.cc | 937 absl::InlinedVector<int64_t, 2> kernel_dims; in HandleConvolution() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | Im2ColFixture.h | 52 …void setup(TensorShape input_shape, DataType data_type, const Size2D &kernel_dims, const PadStride… in setup()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | conv_depthwise_common.h | 39 std::vector<int64_t> kernel_dims; member
|
| H A D | conv_depthwise_backprop_filter_test.cc | 38 std::vector<int64_t> kernel_dims; member
|
| H A D | grouped_convolution_test.cc | 44 std::vector<int64_t> kernel_dims; member
|
| /aosp_15_r20/external/ruy/ruy/ |
| H A D | block_map.h | 97 SidePair<int> kernel_dims; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | eigen_backward_spatial_convolutions.h | 194 DSizes<TensorIndex, 2> kernel_dims; variable
|
| H A D | eigen_backward_cuboid_convolutions.h | 210 DSizes<TensorIndex, 2> kernel_dims; variable
|
| /aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| H A D | TensorConvolution.h | 27 IndexMapper(const InputDims& input_dims, const array<Index, NumKernelDims>& kernel_dims, in IndexMapper()
|
| /aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/ |
| H A D | ClWinogradConv2d.cpp | 51 Size2D winograd_output_tile(const Size2D &input_dims, const Size2D &kernel_dims, DataLayout data_la… in winograd_output_tile()
|
| H A D | ClGemmConv2d.cpp | 449 const Size2D kernel_dims(kernel_width, kernel_height); in validate() local
|
| /aosp_15_r20/external/XNNPACK/test/ |
| H A D | fully-connected.cc | 98 std::vector<size_t> kernel_dims; member in FullyConnectedTestBase
|
| H A D | deconvolution-2d.cc | 103 std::array<size_t, 4> kernel_dims; member in DeconvolutionTestBase
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | randomized_tests.cc | 373 std::vector<int64_t> kernel_dims; member
1784 std::vector<int64_t> input_dims, kernel_dims, stride_dims; in TEST_F() local
2335 std::vector<int64_t> kernel_dims = {d.kernel_dims[0], d.kernel_dims[1], in TEST_F() local
2533 std::vector<int64_t> kernel_dims = {d.kernel_dims[0], d.kernel_dims[1], in TEST_F() local
3421 std::vector<int64_t> input_dims, kernel_dims, stride_dims; in TEST_F() local
|