Searched defs:input_z (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| 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()
|
| 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()
|
| D | runtime_conv_impl.h | 110 Eigen::Index input_y, Eigen::Index input_z, Eigen::Index input_channels, in EigenConv3DImpl()
|
| /external/ComputeLibrary/src/cpu/kernels/depthwiseconv2d/generic/neon/ |
| 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
|