Searched defs:kernel_layout (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/ruy/ruy/ |
| H A D | create_trmul_params.h | 43 const KernelLayout& kernel_layout, in CreatePackedLayout()
55 void CreatePackedMatrix(Side side, const KernelLayout& kernel_layout, in CreatePackedMatrix()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | grouped_convolution_test.cc | 45 std::vector<int64_t> kernel_layout; member
74 std::vector<int64_t> kernel_layout = {3, 2, 1, 0}; in GetConv2DTestCases() local
|
| H A D | conv_depthwise_test.cc | 51 std::vector<int64_t> kernel_layout = {3, 2, 1, 0}; in GetConv2DTestCases() local
|
| H A D | conv_depthwise_common.h | 40 std::vector<int64_t> kernel_layout; member
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | mkldnn_ir.py | 390 kernel_layout, argument
|
| H A D | ir.py | 6147 kernel_layout, argument
|