Searched defs:input_z (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/depthwiseconv2d/generic/neon/ |
| H A D | impl.cpp | 171 const int32_t input_z = id.z() * run_info.conv_stride_y - run_info.conv_pad_top; in depthwise_loop_multiplier1_fp() local
284 const int input_z = id.z() * run_info.conv_stride_y - run_info.conv_pad_top; in depthwise_loop_generic_fp() local
376 const int32_t input_z = id.z() * run_info.conv_stride_y - run_info.conv_pad_top; in depthwise_loop_multiplier1_quantized() local
553 const int32_t input_z = id.z() * run_info.conv_stride_y - run_info.conv_pad_top; in depthwise_loop_generic_quantized() local
667 const int32_t input_z = id.z() * run_info.conv_stride_y - run_info.conv_pad_top; in depthwise_loop_pow2_quantized_per_tensor() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | runtime_conv3d.cc | 27 int64_t input_batch, int64_t input_x, int64_t input_y, int64_t input_z, in __xla_cpu_runtime_EigenConv3DF32()
53 int64_t input_z, int64_t input_channels, int64_t kernel_x, int64_t kernel_y, in __xla_cpu_runtime_EigenConv3DF16()
|
| H A D | runtime_single_threaded_conv3d.cc | 24 int64_t input_batch, int64_t input_x, int64_t input_y, int64_t input_z, in __xla_cpu_runtime_EigenSingleThreadedConv3DF32()
48 int64_t input_z, int64_t input_channels, int64_t kernel_x, int64_t kernel_y, in __xla_cpu_runtime_EigenSingleThreadedConv3DF16()
|
| H A D | runtime_conv_impl.h | 110 Eigen::Index input_y, Eigen::Index input_z, Eigen::Index input_channels, in EigenConv3DImpl()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| H A D | indirection.c | 72 const size_t input_z = output_z * stride_depth + in pytorch_qnnp_indirection_init_conv3d() local
267 const size_t input_z = output_z * stride_depth + in pytorch_qnnp_indirection_init_dwconv() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ReflectionPad.cu | 201 int64_t input_z = ::abs(output_z - pad_front) in parallel_reflection_pad3d() local
|