/aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
H A D | copy_ops_util.cpp | 128 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 D | matmul_ops_util.cpp | 54 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 D | kernel_ops_util.cpp | 230 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 D | slice_util.cpp | 41 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 D | broadcast_util.cpp | 213 Tensor::SizesType* out_sizes, in get_broadcast_target_size() 254 Tensor::SizesType* out_sizes, in get_broadcast_target_size()
|
H A D | padding_util.cpp | 43 Tensor::SizesType* out_sizes, in get_padding_out_target_size()
|
H A D | distance_util.cpp | 24 Tensor::SizesType* out_sizes, in get_pdist_out_target_size()
|
/aosp_15_r20/external/executorch/kernels/test/ |
H A D | op_as_strided_copy_test.cpp | 45 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 D | op_glu_test.cpp | 77 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 D | op_pixel_shuffle_test.cpp | 40 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 D | op_pixel_unshuffle_test.cpp | 40 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 D | qlinear_dynamic.cpp | 117 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 D | qlinear.cpp | 118 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 D | SpectralOps.cpp | 263 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 D | SpectralOps.cpp | 166 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 D | View.cpp | 22 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 D | Full.cpp | 23 std::vector<int64_t> out_sizes; in resize_full_node() local
|
H A D | Arange.cpp | 35 std::vector<int64_t> out_sizes = { in resize_arange_node() local
|
H A D | Upsample.cpp | 25 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 D | op_constant_pad_nd.cpp | 37 IntArrayRef out_sizes, in apply_padding_to_dim() 121 int64_t out_sizes[kTensorDimensionLimit]; in constant_pad_nd_out_impl() local
|
H A D | op_convolution.cpp | 52 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 D | KernelUtils.cpp | 83 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/ |
D | rse_comms_protocol_pointer_access.h | 24 uint32_t out_sizes[PSA_MAX_IOVEC]; member
|
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
H A D | get_item_op.cc | 56 std::vector<int64_t> out_sizes = {}; in Compile() local
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/ |
H A D | qlinear_dynamic.cpp | 122 std::vector<int64_t> out_sizes = input.sizes().vec(); in apply_dynamic_impl() local
|