Searched defs:z_stride (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | runtime_conv3d.cc | 31 int64_t y_stride, int64_t z_stride, int64_t padding_x_before, in __xla_cpu_runtime_EigenConv3DF32()
56 int64_t y_stride, int64_t z_stride, int64_t padding_x_before, in __xla_cpu_runtime_EigenConv3DF16()
|
| H A D | runtime_single_threaded_conv3d.cc | 28 int64_t y_stride, int64_t z_stride, int64_t padding_x_before, in __xla_cpu_runtime_EigenSingleThreadedConv3DF32()
51 int64_t y_stride, int64_t z_stride, int64_t padding_x_before, in __xla_cpu_runtime_EigenSingleThreadedConv3DF16()
|
| H A D | runtime_conv_impl.h | 114 Eigen::Index x_stride, Eigen::Index y_stride, Eigen::Index z_stride, in EigenConv3DImpl()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/pool3d/neon/ |
| H A D | quantized.h | 74 const int z_stride = static_cast<int>(src->info()->strides_in_bytes().z()); in avg_poolingMxNxD_q8_neon_ndhwc() local
255 const int z_stride = static_cast<int>(src->info()->strides_in_bytes().z()); in max_poolingMxNxD_q8_neon_ndhwc() local
|
| H A D | impl.cpp | 64 const int z_stride = static_cast<int>(src->info()->strides_in_bytes().z()); in max_poolingMxNxD_fp_neon_ndhwc() local
176 const int z_stride = static_cast<int>(src->info()->strides_in_bytes().z()); in avg_poolingMxNxD_fp_neon_ndhwc() local
303 const int z_stride = static_cast<int>(src->info()->strides_in_bytes().z()); in l2_poolingMxNxD_fp_neon_ndhwc() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/auxiliary/gallivm/ |
| H A D | lp_bld_sample.c | 2170 LLVMValueRef z_stride, in lp_build_sample_offset()
2221 LLVMValueRef z_stride, in lp_build_tiled_sample_offset()
|
| H A D | lp_bld_sample_aos.c | 759 LLVMValueRef x_stride, y_stride, z_stride; in lp_build_sample_image_linear() local
|
| H A D | lp_bld_sample_soa.c | 134 LLVMValueRef z_stride, in lp_build_sample_texel_soa()
|
| /aosp_15_r20/external/mesa3d/src/nouveau/nil/ |
| H A D | image.rs | 535 let z_stride = u64::from(lvl_ext_B.width * lvl_ext_B.height); in _3d_level_as_2d_array() localVariable
|