Home
last modified time | relevance | path

Searched defs:input_y (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/XNNPACK/src/
H A Dindirection.c56 … const size_t input_y = output_y * stride_height + kernel_y * dilation_height - input_padding_top; in xnn_indirection_init_conv2d() local
119 const size_t input_y = fxdiv_quotient_size_t(y, stride_height_divisor); in xnn_indirection_init_deconv2d() local
175 const size_t input_y = y / stride_height; in xnn_indirection_init_subconv2d() local
227 … const size_t input_y = output_y * stride_height + kernel_y * dilation_height - input_padding_top; in xnn_indirection_init_dwconv2d() local
295 size_t input_y = output_y * stride_height + pooling_y * dilation_height - input_padding_top; in xnn_indirection_init_maxpool2d() local
324 …const size_t input_y = min(doz(output_y * stride_height + pooling_y * dilation_height, input_paddi… in xnn_indirection_init_maxpool2d() local
370 const float input_y = (float) (int32_t) output_y * height_scale; in xnn_indirection_init_resize_bilinear2d_hwc_f16() local
403 float input_y = (float) (int32_t) output_y * height_scale + height_offset; in xnn_indirection_init_resize_bilinear2d_hwc_f16() local
465 const float input_y = (float) (int32_t) output_y * height_scale; in xnn_indirection_init_resize_bilinear2d_hwc_f32() local
498 float input_y = (float) (int32_t) output_y * height_scale + height_offset; in xnn_indirection_init_resize_bilinear2d_hwc_f32() local
[all …]
H A Dim2col.c33 …const size_t input_y = output_y * subsampling_height + kernel_y * dilation_height - input_padding_… in xnn_im2col_conv2d() local
H A Doperator-run.c736 size_t input_y, in xnn_compute_unpooling()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Datan2.cc40 const TfLiteTensor* input_y = tflite::GetInput(context, node, 0); in Atan2Prepare() local
58 TfLiteStatus Atan2(const TfLiteTensor* input_y, in Atan2()
74 const TfLiteTensor* input_y = tflite::GetInput(context, node, 0); in Atan2Eval() local
H A Datan2_custom.cc39 const TfLiteTensor* input_y = tflite::GetInput(context, node, 0); in Atan2Prepare() local
56 TfLiteStatus Atan2(const TfLiteTensor* input_y, const TfLiteTensor* input_x, in Atan2()
71 const TfLiteTensor* input_y = tflite::GetInput(context, node, 0); in Atan2Eval() local
H A Dselect.cc70 const TfLiteTensor* input_y; in SelectPrepare() local
136 const TfLiteTensor* input_y; in SelectEval() local
/aosp_15_r20/external/XNNPACK/test/
H A Dresize-bilinear-operator-tester.h246 const float input_y = (float(output_y) + offset) * height_scale() - offset; in TestNHWCxF16() local
331 const float input_y = (float(output_y) + offset) * height_scale() - offset; in TestNHWCxF32() local
413 const float input_y = (float(output_y) + offset) * height_scale() - offset; in TestNHWCxS8() local
496 const float input_y = (float(output_y) + offset) * height_scale() - offset; in TestNHWCxU8() local
582 const float input_y = (float(output_y) + offset) * height_scale() - offset; in TestNCHWxF32() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dindirection.c76 const size_t input_y = output_y * stride_height + in pytorch_qnnp_indirection_init_conv3d() local
272 const size_t input_y = output_y * stride_height + in pytorch_qnnp_indirection_init_dwconv() local
381 const size_t input_y = y / stride_height; in pytorch_qnnp_indirection_init_deconv2d() local
433 const size_t input_y = in pytorch_qnnp_indirection_init_maxpool2d() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/depthwiseconv2d/generic/neon/
H A Dimpl.cpp170 const int32_t input_y = id.y() * run_info.conv_stride_x - run_info.conv_pad_left; in depthwise_loop_multiplier1_fp() local
283 const int input_y = id.y() * run_info.conv_stride_x - run_info.conv_pad_left; in depthwise_loop_generic_fp() local
375 const int32_t input_y = id.y() * run_info.conv_stride_x - run_info.conv_pad_left; in depthwise_loop_multiplier1_quantized() local
552 const int32_t input_y = id.y() * run_info.conv_stride_x - run_info.conv_pad_left; in depthwise_loop_generic_quantized() local
666 const int32_t input_y = id.y() * run_info.conv_stride_x - run_info.conv_pad_left; in depthwise_loop_pow2_quantized_per_tensor() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_conv3d.cc27 int64_t input_batch, int64_t input_x, int64_t input_y, int64_t input_z, in __xla_cpu_runtime_EigenConv3DF32()
52 Eigen::half* rhs, int64_t input_batch, int64_t input_x, int64_t input_y, in __xla_cpu_runtime_EigenConv3DF16()
H A Druntime_single_threaded_conv3d.cc24 int64_t input_batch, int64_t input_x, int64_t input_y, int64_t input_z, in __xla_cpu_runtime_EigenSingleThreadedConv3DF32()
47 Eigen::half* rhs, int64_t input_batch, int64_t input_x, int64_t input_y, in __xla_cpu_runtime_EigenSingleThreadedConv3DF16()
H A Druntime_conv_impl.h33 Eigen::Index input_y, Eigen::Index input_channels, Eigen::Index kernel_x, in EigenConv2DImpl()
110 Eigen::Index input_y, Eigen::Index input_z, Eigen::Index input_channels, in EigenConv3DImpl()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dresize_bilinear.h92 float input_y; in ResizeBilinear() local
184 int32_t input_y, y0, y1; in ResizeBilinearInteger() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReflectionPad.cu84 auto input_y = ::abs(output_y - pad_t) in get_index_mapping2d() local
195 int64_t input_y = ::abs(output_y - pad_top) in parallel_reflection_pad3d() local
H A DUpSampleBicubic2d.cu139 int input_y = floorf(real_y); in upsample_bicubic2d_backward_out_frame() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/
H A Dconvolver.hpp121 …int input_y = (output_y * m_convolver.m_params.output_stride_h) + m_convolver.m_kernel_y[m_current… in next_block() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSampleBicubic2d.cpp149 int64_t input_y; in upsample_bicubic2d_backward_out_frame() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dresize_bilinear.h1523 float input_y; in ResizeBilinearGeneric() local
1575 float input_y; in ResizeBilinearGenericSmallChannel() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dfunction_optimizer_test.cc511 const string input_y = "Func/c/input/_1"; in TEST_F() local
773 const string input_y = "Func/c/input/_1"; in TEST_F() local
H A Dconstant_folding.cc3427 const string input_y = input_x == ctx.op_child->input(0) in ConstantPushDown() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dfunctions_test.cc598 const NodeDef &input_y = item.function_body().node(1); in TEST_F() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/internal/
H A DCpuGemmAssemblyDispatch.cpp273 … int64_t input_y = (output_y * _cp.output_stride_h) + kernel_y - _cp.padding_top; in prepare_indirect_buffer() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/
H A Dpass.cc2535 Operation *input_y = left_leaf_is_const in matchAndRewrite() local