Searched defs:kernel_z (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | runtime_conv3d.cc | 29 int64_t kernel_z, int64_t kernel_channels, int64_t kernel_filters, in __xla_cpu_runtime_EigenConv3DF32()
54 int64_t kernel_z, int64_t kernel_channels, int64_t kernel_filters, in __xla_cpu_runtime_EigenConv3DF16()
|
| H A D | runtime_single_threaded_conv3d.cc | 26 int64_t kernel_z, int64_t kernel_channels, int64_t kernel_filters, in __xla_cpu_runtime_EigenSingleThreadedConv3DF32()
49 int64_t kernel_z, int64_t kernel_channels, int64_t kernel_filters, in __xla_cpu_runtime_EigenSingleThreadedConv3DF16()
|
| H A D | runtime_conv_impl.h | 111 Eigen::Index kernel_x, Eigen::Index kernel_y, Eigen::Index kernel_z, in EigenConv3DImpl()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| H A D | depthwise_conv.h | 191 const int kernel_z = weights.shape.d; in RearrangeWeightsForDWConv3D() local
223 const int kernel_z = weights.shape.d; in UploadWeightsForDWConv3D() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| H A D | indirection.c | 71 for (size_t kernel_z = 0; kernel_z < kernel_depth; kernel_z++) { in pytorch_qnnp_indirection_init_conv3d() local
266 for (size_t kernel_z = 0; kernel_z < kernel_depth; kernel_z++) { in pytorch_qnnp_indirection_init_dwconv() local
|