Home
last modified time | relevance | path

Searched defs:out_sizes (Results 1 – 25 of 64) sorted by relevance

123

/aosp_15_r20/external/executorch/kernels/portable/cpu/util/
H A Dcopy_ops_util.cpp128 exec_aten::SizesType* out_sizes, in get_cat_out_target_size()
308 exec_aten::SizesType* out_sizes, in get_permute_copy_out_target_size()
346 exec_aten::SizesType* out_sizes, in get_pixel_shuffle_out_target_size()
368 exec_aten::SizesType* out_sizes, in get_pixel_unshuffle_out_target_size()
403 exec_aten::SizesType* out_sizes, in get_select_copy_out_target_size()
447 exec_aten::SizesType* out_sizes, in get_split_with_sizes_copy_out_target_size()
470 exec_aten::SizesType* out_sizes, in get_squeeze_copy_dim_out_target_size()
523 exec_aten::SizesType* out_sizes, in get_squeeze_copy_dims_out_target_size()
589 exec_aten::SizesType* out_sizes, in get_stack_out_target_size()
879 exec_aten::SizesType* out_sizes) { in get_view_copy_target_size()
[all …]
H A Dmatmul_ops_util.cpp54 Tensor::SizesType* out_sizes, in get_bmm_out_target_size()
90 Tensor::SizesType* out_sizes, in get_mm_out_target_size()
100 Tensor::SizesType* out_sizes, in get_linear_out_target_size()
H A Dkernel_ops_util.cpp230 exec_aten::SizesType* out_sizes, in calculate_kernel_output_sizes()
303 exec_aten::SizesType* const out_sizes, in get_avg_pool2d_out_target_size()
415 exec_aten::SizesType* out_sizes, in get_convolution_out_target_size()
506 exec_aten::SizesType* out_sizes, in get_max_pool2d_with_indices_out_target_size()
H A Dslice_util.cpp41 exec_aten::SizesType* out_sizes, in get_narrow_copy_out_target_size()
68 exec_aten::SizesType* out_sizes, in get_slice_copy_out_target_size()
H A Dbroadcast_util.cpp213 Tensor::SizesType* out_sizes, in get_broadcast_target_size()
254 Tensor::SizesType* out_sizes, in get_broadcast_target_size()
H A Dpadding_util.cpp43 Tensor::SizesType* out_sizes, in get_padding_out_target_size()
H A Ddistance_util.cpp24 Tensor::SizesType* out_sizes, in get_pdist_out_target_size()
/aosp_15_r20/external/executorch/kernels/test/
H A Dop_as_strided_copy_test.cpp45 const std::vector<int32_t> out_sizes = {2, 2, 2}; in test_detach_copy_out() local
77 const std::vector<int32_t> out_sizes = {2, 2, 2}; in test_as_strided_copy_out_invalid_parameters() local
146 const std::vector<int32_t> out_sizes = {2, 2, 2}; in test_detach_copy_out() local
170 const std::vector<int32_t> out_sizes = {2, 2, 2}; in test_detach_copy_out() local
213 const std::vector<int32_t> out_sizes = {2, 2, 2}; in TEST_F() local
H A Dop_glu_test.cpp77 const std::vector<int32_t> out_sizes = {1, 2}; in test_glu_out_invalid_dim() local
95 const std::vector<int32_t> out_sizes = {1, 2}; in test_div_invalid_input_dtype_dies() local
109 const std::vector<int32_t> out_sizes = {1, 2}; in test_div_invalid_output_dtype_dies() local
134 const std::vector<int32_t> out_sizes = {4, 1}; in TEST_F() local
H A Dop_pixel_shuffle_test.cpp40 const std::vector<int32_t> out_sizes = {1, 1, 4, 4}; in test_pixel_shuffle() local
81 const std::vector<int32_t> out_sizes = {1, 4, 1, 1, 4, 4}; in TEST_F() local
H A Dop_pixel_unshuffle_test.cpp40 const std::vector<int32_t> out_sizes = {1, 4, 2, 2}; in test_pixel_unshuffle() local
80 const std::vector<int32_t> out_sizes = {1, 4, 1, 4, 2, 2}; in TEST_F() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqlinear_dynamic.cpp117 std::vector<int64_t> out_sizes = input.sizes().vec(); in apply_dynamic_impl() local
355 std::vector<int64_t> out_sizes = input.sizes().vec(); in apply_dynamic_impl() local
598 auto out_sizes = input.sizes().vec(); in apply_dynamic_impl() local
713 auto out_sizes = input.sym_sizes().vec(); in meta() local
777 auto out_sizes = input.sym_sizes().vec(); in wrapped_fbgemm_linear_fp16_weight_meta() local
H A Dqlinear.cpp118 at::DimVector out_sizes(input.sizes()); in apply_impl() local
354 std::vector<int64_t> out_sizes = input.sizes().vec(); in apply_with_input_q_dq_qweight_dq_output_fp32_impl() local
554 std::vector<int64_t> out_sizes = input.sizes().vec(); in apply_impl_xnnp() local
697 std::vector<int64_t> out_sizes = input.sizes().vec(); in apply_impl() local
871 auto out_sizes = input.sizes().vec(); in apply_impl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkl/
H A DSpectralOps.cpp263 DimVector out_sizes(in_sizes.begin(), in_sizes.end()); in _fft_c2r_mkl() local
283 DimVector out_sizes(input_sizes.begin(), input_sizes.end()); in _fft_r2c_mkl() local
430 static Tensor& _exec_fft(Tensor& out, const Tensor& self, IntArrayRef out_sizes, in _exec_fft()
533 DimVector out_sizes(in_sizes.begin(), in_sizes.end()); in _fft_c2r_mkl() local
543 DimVector out_sizes(input_sizes.begin(), input_sizes.end()); in _fft_r2c_mkl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DSpectralOps.cpp166 static const Tensor& _exec_fft(Tensor& out, const Tensor& self, IntArrayRef out_sizes, in _exec_fft()
338 IntArrayRef out_sizes = onesided ? onesided_sizes : input_sizes; in _fft_r2c_cufft() local
421 DimVector out_sizes(in_sizes.begin(), in_sizes.end()); in _fft_c2r_cufft() local
464 auto out_sizes = self.sizes(); in _fft_c2c_cufft() local
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DView.cpp22 std::vector<int64_t> out_sizes(view_sizes.begin(), view_sizes.end()); in compute_out_sizes() local
52 std::vector<int64_t> out_sizes = in resize_view_node() local
H A DFull.cpp23 std::vector<int64_t> out_sizes; in resize_full_node() local
H A DArange.cpp35 std::vector<int64_t> out_sizes = { in resize_arange_node() local
H A DUpsample.cpp25 std::vector<int64_t> out_sizes = self->sizes(); // NCHW in resize_upsample_nearest2d_node() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_constant_pad_nd.cpp37 IntArrayRef out_sizes, in apply_padding_to_dim()
121 int64_t out_sizes[kTensorDimensionLimit]; in constant_pad_nd_out_impl() local
H A Dop_convolution.cpp52 SizesArrayRef out_sizes, in conv2d_impl()
205 SizesArrayRef out_sizes = out.sizes(); in convolution_wrapper() local
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/utils/
H A DKernelUtils.cpp83 std::vector<int64_t> out_sizes(2); in calc_out_sizes_hw() local
126 std::vector<int64_t> out_sizes(2); in calc_transpose_out_sizes_hw() local
/aosp_15_r20/external/trusty/arm-trusted-firmware/drivers/arm/rse/
Drse_comms_protocol_pointer_access.h24 uint32_t out_sizes[PSA_MAX_IOVEC]; member
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/
H A Dget_item_op.cc56 std::vector<int64_t> out_sizes = {}; in Compile() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/
H A Dqlinear_dynamic.cpp122 std::vector<int64_t> out_sizes = input.sizes().vec(); in apply_dynamic_impl() local

123