Searched defs:shape2d (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/ |
| H A D | ClDirectConv2dKernel.cpp | 425 const TensorShape shape2d(image_w, image_h); in run_op() local
436 const TensorShape shape2d(image_w, image_h); in run_op() local
447 const TensorShape shape2d(image_w, image_h); in run_op() local
|
| H A D | ClIndirectConv2dKernel.cpp | 281 const TensorShape shape2d(image_w, image_h); in run_op() local
|
| H A D | ClGemmMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp | 325 …const TensorShape shape2d(src1->info()->dimension(0) / 4, src1->info()->dimension(1) * src1->info(… in run_op() local
|
| H A D | ClGemmMatrixMultiplyReshapedKernel.cpp | 360 …const TensorShape shape2d(src1->info()->dimension(0) / 4, src1->info()->dimension(1) * src1->info(… in run_op() local
|
| H A D | ClGemmMatrixMultiplyReshapedOnlyRhsKernel.cpp | 375 …const TensorShape shape2d(src1->info()->dimension(0) / 4, src1->info()->dimension(1) * src1->info(… in run_op() local
|
| /aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/runtime/gpu/cl/ |
| H A D | ClKernelRuntime.cpp | 90 …const TensorShape shape2d(tensor->info()->dimension(0) / 4, tensor->info()->dimension(1) * tensor-… in add_tensor_argument() local
106 …const TensorShape shape2d(tensor->info()->dimension(0) / 4, tensor->info()->dimension(1) * tensor-… in add_tensor_argument() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/ |
| H A D | CLDepthwiseConvolutionLayerNativeKernel.cpp | 378 const TensorShape shape2d(image_w, image_h); in run() local
387 const TensorShape shape2d(image_w, image_h); in run() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/CL/ |
| H A D | CLUtils.cpp | 35 …_from_buffer(const cl::Context &ctx, const cl::Buffer &buffer, const TensorShape &shape2d, DataTyp… in create_image2d_from_buffer()
|