Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
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
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
Dselect.cc70 const TfLiteTensor* input_y; in SelectPrepare() local
136 const TfLiteTensor* input_y; in SelectEval() local
/external/XNNPACK/src/
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 …]
Dim2col.c33 …const size_t input_y = output_y * subsampling_height + kernel_y * dilation_height - input_padding_… in xnn_im2col_conv2d() local
Doperator-run.c736 size_t input_y, in xnn_compute_unpooling()
/external/XNNPACK/test/
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
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
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()
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()
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()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dresize_bilinear.h92 float input_y; in ResizeBilinear() local
184 int32_t input_y, y0, y1; in ResizeBilinearInteger() local
/external/ComputeLibrary/src/cpu/kernels/depthwiseconv2d/generic/neon/
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
/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/
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
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions_test.cc598 const NodeDef &input_y = item.function_body().node(1); in TEST_F() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dresize_bilinear.h1523 float input_y; in ResizeBilinearGeneric() local
1575 float input_y; in ResizeBilinearGenericSmallChannel() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
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
Dconstant_folding.cc3427 const string input_y = input_x == ctx.op_child->input(0) in ConstantPushDown() local
/external/ComputeLibrary/src/cpu/operators/internal/
DCpuGemmAssemblyDispatch.cpp273 … int64_t input_y = (output_y * _cp.output_stride_h) + kernel_y - _cp.padding_top; in prepare_indirect_buffer() local
/external/tensorflow/tensorflow/core/transforms/constant_folding/
Dpass.cc2535 Operation *input_y = left_leaf_is_const in matchAndRewrite() local