Home
last modified time | relevance | path

Searched refs:resize_tensor (Results 1 – 25 of 143) sorted by relevance

123456

/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_native_batch_norm.cpp42 resize_tensor(out, in.sizes()) == Error::Ok, in _native_batch_norm_legit_no_training_out()
47 ctx, resize_tensor(mean_out, {0}) == Error::Ok, InvalidArgument, ret_val); in _native_batch_norm_legit_no_training_out()
51 resize_tensor(invstd_out, {0}) == Error::Ok, in _native_batch_norm_legit_no_training_out()
246 resize_tensor(out, in.sizes()) == Error::Ok, in _native_batch_norm_legit_no_stats_out()
252 resize_tensor(mean_out, {static_cast<SizesType>(C)}) == Error::Ok, in _native_batch_norm_legit_no_stats_out()
258 resize_tensor(invstd_out, {static_cast<SizesType>(C)}) == Error::Ok, in _native_batch_norm_legit_no_stats_out()
H A Dop_native_layer_norm.cpp154 resize_tensor(out, input.sizes()) == Error::Ok, in native_layer_norm_out()
160 resize_tensor(mean_out, {mean_rstd_sizes, mean_rstd_ndim}) == Error::Ok, in native_layer_norm_out()
166 resize_tensor(rstd_out, {mean_rstd_sizes, mean_rstd_ndim}) == Error::Ok, in native_layer_norm_out()
H A Dop_native_group_norm.cpp145 resize_tensor(out, input.sizes()) == Error::Ok, in native_group_norm_out()
151 resize_tensor(mean_out, {mean_rstd_sizes, 2}) == Error::Ok, in native_group_norm_out()
157 resize_tensor(rstd_out, {mean_rstd_sizes, 2}) == Error::Ok, in native_group_norm_out()
H A Dop_t_copy.cpp37 ctx, resize_tensor(out, in.sizes()) == Error::Ok, InvalidArgument, out); in t_copy_out()
62 resize_tensor(out, {expected_out_size, expected_out_dim}) == Error::Ok, in t_copy_out()
H A Dop_convolution_backward.cpp272 resize_tensor(grad_input, input.sizes()) == Error::Ok, in convolution_backward_out()
278 resize_tensor(grad_weight, weight.sizes()) == Error::Ok, in convolution_backward_out()
285 resize_tensor(grad_bias, bias_sizes_opt.value()) == Error::Ok, in convolution_backward_out()
H A Dop_max_pool2d_with_indices.cpp62 resize_tensor(out, {output_sizes, output_ndim}) == Error::Ok, in max_pool2d_with_indices_out()
68 resize_tensor(indices, {output_sizes, output_ndim}) == Error::Ok, in max_pool2d_with_indices_out()
H A Dop_squeeze_copy.cpp47 resize_tensor(out, {expected_out_size, expected_out_dim}) == Error::Ok, in squeeze_copy_dim_out()
81 resize_tensor(out, {expected_out_size, expected_out_dim}) == Error::Ok, in squeeze_copy_dims_out()
H A Dop_arange.cpp38 resize_tensor(out, {&out_length, 1}) == Error::Ok, in arange_out()
87 resize_tensor(out, {&out_length, 1}) == Error::Ok, in arange_start_out()
H A Dop_fill.cpp40 resize_tensor(out, a.sizes()) == Error::Ok, in fill_scalar_out()
85 resize_tensor(out, a.sizes()) == Error::Ok, in fill_tensor_out()
H A Dop_index.cpp47 ctx, resize_tensor(out, in.sizes()) == Error::Ok, InvalidArgument, out); in index_Tensor_out()
72 resize_tensor(out, {expected_size, expected_ndim}) == Error::Ok, in index_Tensor_out()
H A Dop_masked_select.cpp42 ctx, resize_tensor(out, {0}) == Error::Ok, InvalidArgument, out); in masked_select_out()
73 ctx, resize_tensor(out, {out_numel}) == Error::Ok, InvalidArgument, out); in masked_select_out()
H A Dop_index_put.cpp50 ctx, resize_tensor(out, in.sizes()) == Error::Ok, InvalidArgument, out); in index_put_out()
89 ctx, resize_tensor(out, in.sizes()) == Error::Ok, InvalidArgument, out); in index_put_out()
H A Dop_min.cpp56 resize_tensor(min_indices, min.sizes()) == Error::Ok, in min_out()
110 ctx, resize_tensor(out, {}) == Error::Ok, InvalidArgument, out); in min_unary_out()
H A Dop_pow.cpp94 ctx, resize_tensor(out, a.sizes()) == Error::Ok, InvalidArgument, out); in pow_Tensor_Scalar_out()
141 ctx, resize_tensor(out, b.sizes()) == Error::Ok, InvalidArgument, out); in pow_Scalar_out()
H A Dop_max.cpp56 resize_tensor(max_indices, max.sizes()) == Error::Ok, in max_out()
110 ctx, resize_tensor(out, {}) == Error::Ok, InvalidArgument, out); in max_unary_out()
/aosp_15_r20/external/executorch/runtime/core/exec_aten/testing_util/test/
H A Dtensor_factory_test.cpp30 using executorch::runtime::resize_tensor;
41 EXPECT_EQ(resize_tensor(t, ArrayRef<SizesType>({1, 1})), Error::Ok); in resize_tensor_to_assert_static()
43 EXPECT_NE(resize_tensor(t, ArrayRef<SizesType>({1, 1})), Error::Ok); in resize_tensor_to_assert_static()
54 EXPECT_EQ(resize_tensor(t, ArrayRef<SizesType>({1, 1})), Error::Ok); in resize_tensor_to_assert_dynamic_bound()
55 EXPECT_EQ(resize_tensor(t, ArrayRef<SizesType>({100, 100})), Error::Ok); in resize_tensor_to_assert_dynamic_bound()
57 EXPECT_EQ(resize_tensor(t, ArrayRef<SizesType>({1, 1})), Error::Ok); in resize_tensor_to_assert_dynamic_bound()
58 EXPECT_NE(resize_tensor(t, ArrayRef<SizesType>({100, 100})), Error::Ok); in resize_tensor_to_assert_dynamic_bound()
68 EXPECT_EQ(resize_tensor(t, ArrayRef<SizesType>({1, 1})), Error::Ok); in resize_tensor_to_assert_dynamic_unbound()
71 EXPECT_EQ(resize_tensor(t, ArrayRef<SizesType>({100, 100})), Error::Ok); in resize_tensor_to_assert_dynamic_unbound()
74 EXPECT_NE(resize_tensor(t, ArrayRef<SizesType>({100, 100})), Error::Ok); in resize_tensor_to_assert_dynamic_unbound()
[all …]
/aosp_15_r20/external/executorch/kernels/optimized/cpu/
H A Dop_native_layer_norm.cpp142 resize_tensor(out, input.sizes()) == Error::Ok, in opt_native_layer_norm_out()
148 resize_tensor(mean_out, {mean_rstd_sizes, mean_rstd_ndim}) == Error::Ok, in opt_native_layer_norm_out()
154 resize_tensor(rstd_out, {mean_rstd_sizes, mean_rstd_ndim}) == Error::Ok, in opt_native_layer_norm_out()
H A Dop_mul.cpp88 auto error = resize_tensor(out, lhs->sizes()); in handle_last_dim_broadcast()
142 auto error = resize_tensor(out, lhs->sizes()); in handle_broadcast_mul()
223 auto error = resize_tensor(out, a.sizes()); in opt_mul_out()
298 auto error = resize_tensor(out, a.sizes()); in opt_mul_scalar_out()
H A Dop_div.cpp105 auto error = resize_tensor(out, a.sizes()); in opt_div_out()
137 auto error = resize_tensor(out, lhs->sizes()); in opt_div_out()
214 auto error = resize_tensor(out, a.sizes()); in opt_div_scalar_out()
H A Dop_add.cpp121 auto error = resize_tensor(out, a.sizes()); in opt_add_out()
157 auto error = resize_tensor(out, lhs->sizes()); in opt_add_out()
234 auto error = resize_tensor(out, a.sizes()); in opt_add_scalar_out()
H A Dop_sub.cpp144 auto error = resize_tensor(out, a.sizes()); in opt_sub_out()
180 auto error = resize_tensor(out, lhs->sizes()); in opt_sub_out()
268 auto error = resize_tensor(out, a.sizes()); in opt_sub_scalar_out()
/aosp_15_r20/external/executorch/kernels/prim_ops/
H A Det_copy_index.cpp19 using torch::executor::resize_tensor;
101 resize_tensor(copy_to, {expected_output_size, copy_to.sizes().size()}); in et_copy_index()
H A Det_view.cpp21 using torch::executor::resize_tensor;
82 resize_tensor( in et_view()
/aosp_15_r20/external/executorch/kernels/test/custom_kernel_example/
H A Dop_relu.cpp22 using executorch::runtime::resize_tensor;
44 auto error = resize_tensor( in relu()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dtopk_v2.cc73 auto resize_tensor = [context](TfLiteTensor* tensor, TfLiteIntArray* new_size, in ResizeOutput() local
83 TF_LITE_ENSURE_OK(context, resize_tensor(output_indexes, output_indexes_shape, in ResizeOutput()
86 resize_tensor(output_values, output_values_shape, nullptr)); in ResizeOutput()

123456