Searched defs:output_h (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ReflectionPad.cpp | 133 int64_t output_h = input_h + pad_top + pad_bottom; in TORCH_META_FUNC() local
194 int64_t output_h = input_h + pad_top + pad_bottom; in TORCH_META_FUNC() local
238 int64_t output_h = input_h + pad_t + pad_b; in reflection_pad2d_out_template() local
288 int64_t output_h = input_h + pad_t + pad_b; in reflection_pad2d_backward_out_template() local
|
| /aosp_15_r20/external/armnn/third-party/stb/ |
| H A D | stb_image_resize.h | 503 int output_h; member
2168 int output_h = stbir_info->output_h; in stbir__buffer_loop_downsample() local
2205 static void stbir__setup(stbir__info *info, int input_w, int input_h, int output_w, int output_h, i… in stbir__setup()
2429 void* output_data, int output_w, int output_h, int output_stride_in_bytes, in stbir__resize_arbitrary()
2461 … unsigned char *output_pixels, int output_w, int output_h, int output_stride_in_bytes, in stbir_resize_uint8()
2471 … float *output_pixels, int output_w, int output_h, int output_stride_in_bytes, in stbir_resize_float()
2481 … unsigned char *output_pixels, int output_w, int output_h, int output_stride_in_bytes, in stbir_resize_uint8_srgb()
2491 … unsigned char *output_pixels, int output_w, int output_h, int output_stride_in_bytes, in stbir_resize_uint8_srgb_edgemode()
2502 … unsigned char *output_pixels, int output_w, int output_h, int output_stride_in_bytes, in stbir_resize_uint8_generic()
2514 … stbir_uint16 *output_pixels , int output_w, int output_h, int output_stride_in_bytes, in stbir_resize_uint16_generic()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | cudnn_rnn_ops.cc | 797 Tensor* output, Tensor* output_h, Tensor* output_c, in DoForward()
876 const Tensor* output, const Tensor* output_h, const Tensor* output_c, in DoBackward()
1526 Tensor* output_h = nullptr; in ComputeAndReturnAlgorithm() local
1573 Tensor* output, Tensor* output_h, in MaybeAutotune()
1589 Tensor** output, Tensor** output_h, in AllocateOutputs()
1679 const Tensor* params, Tensor* output, Tensor* output_h, in MaybeAutotune()
1928 const Tensor* output_h = nullptr; in ComputeImpl() local
1988 const Tensor** output_h, const Tensor** output_c, in ExtractBackwardInputs()
|
| H A D | softmax_op_gpu.cu.cc | 128 Eigen::half* output_h = reinterpret_cast<Eigen::half*>(&output_d); in GenerateNormalizedProb() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ReflectionPad.cu | 310 int output_h = input_h + pad_t + pad_b; in reflection_pad2d_out_template() local
390 int output_h = input_h + pad_t + pad_b; in reflection_pad2d_backward_out_template() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/profiler/ |
| H A D | util.cpp | 642 uint64_t output_h = in computeFlops() local
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/ |
| H A D | ShapeCalculator.h | 772 const unsigned int output_h = DIV_CEIL(output_conv2d_shape[1] * output_conv2d_shape[2], m0); in compute_indirect_buffer_shape() local
|