| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conv_depthwise_common.cc | 44 absl::StrJoin(spec.kernel_layout, "_"), "_output_dims_", in DepthwiseConvolution2DTestDataToString()
77 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextDepthwiseConvolution2D()
83 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextDepthwiseConvolution2D()
101 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextDepthwiseConvolution2D()
107 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextDepthwiseConvolution2D()
125 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextDepthwiseConvolution2D()
131 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextDepthwiseConvolution2D()
|
| 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
86 config.kernel_layout = {3, 2, 1, 0}; in GetConv2DTestCases()
135 absl::StrJoin(spec.kernel_layout, "_"), "_output_dims_", in GroupedConvolution2DTestDataToString()
168 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextGroupedConvolution2D()
174 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextGroupedConvolution2D()
194 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextGroupedConvolution2D()
200 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextGroupedConvolution2D()
219 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextGroupedConvolution2D()
225 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextGroupedConvolution2D()
|
| D | conv_depthwise_test.cc | 51 std::vector<int64_t> kernel_layout = {3, 2, 1, 0}; in GetConv2DTestCases() local
60 config.kernel_layout = {3, 2, 1, 0}; in GetConv2DTestCases()
|
| D | conv_depthwise_common.h | 40 std::vector<int64_t> kernel_layout; member
|
| /external/pytorch/torch/_inductor/ |
| D | mkldnn_ir.py | 168 kernel_layout = FixedLayout(
182 return inputs, constant_args, kernel_layout, req_stride_order
214 kernel_layout = FixedLayout(
226 return inputs, constant_args, kernel_layout, req_stride_order
284 (inputs, constant_args, kernel_layout, _) = _prepare_convolution_fusion_create(
293 layout=kernel_layout,
366 kernel_layout,
381 layout=kernel_layout,
390 kernel_layout, argument
398 kernel_layout,
[all …]
|
| D | ir.py | 5924 kernel_layout = FixedLayout(
5938 return inputs, constant_args, kernel_layout, req_stride_order
5970 kernel_layout = FixedLayout(
5982 return inputs, constant_args, kernel_layout, req_stride_order
6040 (inputs, constant_args, kernel_layout, _) = _prepare_convolution_fusion_create(
6049 layout=kernel_layout,
6123 kernel_layout,
6138 layout=kernel_layout,
6147 kernel_layout, argument
6155 kernel_layout,
[all …]
|
| /external/ruy/ruy/ |
| D | create_trmul_params.h | 43 const KernelLayout& kernel_layout, in CreatePackedLayout() argument
48 packed_layout->rows = round_up_pot(src.rows, kernel_layout.rows); in CreatePackedLayout()
49 packed_layout->cols = round_up_pot(src.cols, kernel_layout.cols); in CreatePackedLayout()
51 packed_layout->kernel = kernel_layout; in CreatePackedLayout()
55 void CreatePackedMatrix(Side side, const KernelLayout& kernel_layout, in CreatePackedMatrix() argument
70 CreatePackedLayout(src.layout, kernel_layout, &packed_matrix->layout); in CreatePackedMatrix()
|
| /external/pytorch/aten/src/ATen/native/vulkan/api/ |
| D | Context.cpp | 58 shader_layout_cache().retrieve(shader_descriptor.kernel_layout); in get_descriptor_set()
71 shader_layout, shader_descriptor.kernel_layout); in get_descriptor_set()
|
| D | Shader.cpp | 30 kernel_layout{std::move(layout)} {}
45 kernel_layout{std::move(layout)},
|
| D | Shader.h | 56 ShaderLayout::Signature kernel_layout{};
|
| /external/executorch/backends/vulkan/runtime/api/ |
| D | Context.cpp | 95 shader_layout_cache().retrieve(shader_descriptor.kernel_layout); in get_descriptor_set()
115 shader_layout, shader_descriptor.kernel_layout); in get_descriptor_set()
|
| /external/executorch/backends/vulkan/runtime/graph/ops/glsl/ |
| D | conv1d.glsl | 41 ${layout_declare_spec_const(C, "int", "kernel_layout", "DEFAULT_LAYOUT")}
42 const lowp ivec4 kernel_axis_map = unhash_axis_map(kernel_layout);
|
| /external/executorch/backends/vulkan/runtime/vk_api/ |
| D | Shader.h | 61 ShaderLayout::Signature kernel_layout{};
|
| D | Shader.cpp | 37 kernel_layout{std::move(layout)},
|
| /external/executorch/backends/vulkan/runtime/graph/ |
| D | ComputeGraph.cpp | 455 for (const VkDescriptorType arg_type : shader_info.kernel_layout) { in update_descriptor_counts()
|