Home
last modified time | relevance | path

Searched defs:output_z (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_conv3d.cc30 int64_t output_x, int64_t output_y, int64_t output_z, int64_t x_stride, in __xla_cpu_runtime_EigenConv3DF32()
55 int64_t output_x, int64_t output_y, int64_t output_z, int64_t x_stride, in __xla_cpu_runtime_EigenConv3DF16()
H A Druntime_single_threaded_conv3d.cc27 int64_t output_x, int64_t output_y, int64_t output_z, int64_t x_stride, in __xla_cpu_runtime_EigenSingleThreadedConv3DF32()
50 int64_t output_x, int64_t output_y, int64_t output_z, int64_t x_stride, in __xla_cpu_runtime_EigenSingleThreadedConv3DF16()
H A Druntime_conv_impl.h113 Eigen::Index output_x, Eigen::Index output_y, Eigen::Index output_z, in EigenConv3DImpl()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dindirection.c67 const size_t output_z = z_yx.quotient; in pytorch_qnnp_indirection_init_conv3d() local
265 for (size_t output_z = 0; output_z < output_depth; output_z++) { in pytorch_qnnp_indirection_init_dwconv() local
H A Dconv-run.cc295 size_t output_z) { in compute_dwconv3d_multiipass()
H A Doperator-run.c496 size_t output_z) { in compute_dwconv3d_multiipass()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dfunction_optimizer_test.cc512 const string output_z = "Func/c/output/_2"; in TEST_F() local
774 const string output_z = "Func/c/output/_2"; in TEST_F() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReflectionPad.cu181 int64_t output_z = output_id / (output.size(3) * output.size(4)); in parallel_reflection_pad3d() local