| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_min_test.cpp | 40 TensorFactory<IN_DTYPE> tf_in; in test_min_out_invalid_dimensions() local
43 Tensor in = tf_in.ones(/*sizes=*/{2, 3, 4}); in test_min_out_invalid_dimensions()
44 Tensor min = tf_in.zeros({2, 3, 2}); in test_min_out_invalid_dimensions()
45 Tensor min_indices = tf_in.zeros({2, 3}); in test_min_out_invalid_dimensions()
53 min = tf_in.zeros({2, 3, 2}); in test_min_out_invalid_dimensions()
54 min_indices = tf_in.zeros({2, 3, 2}); in test_min_out_invalid_dimensions()
60 min = tf_in.zeros({2, 3, 1}); in test_min_out_invalid_dimensions()
61 min_indices = tf_in.zeros({2, 3, 1}); in test_min_out_invalid_dimensions()
67 min = tf_in.zeros({2, 3, 1}); in test_min_out_invalid_dimensions()
68 min_indices = tf_in.zeros({2, 3, 1}); in test_min_out_invalid_dimensions()
[all …]
|
| H A D | op_max_test.cpp | 40 TensorFactory<IN_DTYPE> tf_in; in test_max_out_invalid_dimensions() local
43 Tensor self = tf_in.ones(/*sizes=*/{2, 3, 4}); in test_max_out_invalid_dimensions()
44 Tensor max = tf_in.zeros({2, 3, 2}); in test_max_out_invalid_dimensions()
45 Tensor max_indices = tf_in.zeros({2, 3}); in test_max_out_invalid_dimensions()
53 max = tf_in.zeros({2, 3, 2}); in test_max_out_invalid_dimensions()
54 max_indices = tf_in.zeros({2, 3, 2}); in test_max_out_invalid_dimensions()
60 max = tf_in.zeros({2, 3, 1}); in test_max_out_invalid_dimensions()
61 max_indices = tf_in.zeros({2, 3, 1}); in test_max_out_invalid_dimensions()
67 max = tf_in.zeros({2, 3, 1}); in test_max_out_invalid_dimensions()
68 max_indices = tf_in.zeros({2, 3, 1}); in test_max_out_invalid_dimensions()
[all …]
|
| H A D | op_glu_test.cpp | 60 TensorFactory<INPUT_DTYPE> tf_in; in test_glu_out_mismatched_shape() local
63 Tensor in = tf_in.zeros(/*sizes=*/{4, 4, 4}); in test_glu_out_mismatched_shape()
64 Tensor out = tf_in.zeros(/*sizes=*/{2, 4, 2}); in test_glu_out_mismatched_shape()
68 out = tf_in.zeros(/*sizes=*/{4, 4, 4}); in test_glu_out_mismatched_shape()
75 TensorFactory<INPUT_DTYPE> tf_in; in test_glu_out_invalid_dim() local
76 Tensor in = tf_in.zeros(/*sizes=*/{2, 2}); in test_glu_out_invalid_dim()
78 Tensor out = tf_in.zeros(out_sizes); in test_glu_out_invalid_dim()
84 in = tf_in.zeros(/*sizes=*/{3, 2}); in test_glu_out_invalid_dim()
91 TensorFactory<INPUT_DTYPE> tf_in; in test_div_invalid_input_dtype_dies() local
96 Tensor in = tf_in.ones(sizes); in test_div_invalid_input_dtype_dies()
|
| H A D | op_to_copy_test.cpp | 98 TensorFactory<INPUT_DTYPE> tf_in; in test_runner_static_cast() local
105 Tensor input = tf_in.make(test_case.sizes, data_in); in test_runner_static_cast()
127 TensorFactory<INPUT_DTYPE> tf_in; in test_runner_to_bool() local
132 Tensor input = tf_in.make({(int)test_case.size()}, data_in); in test_runner_to_bool()
155 TensorFactory<ScalarType::Bool> tf_in; in test_runner_from_bool() local
160 Tensor input = tf_in.make({(int)test_case.size()}, test_case); in test_runner_from_bool()
187 TensorFactory<INPUT_DTYPE> tf_in; in test_runner_hardcode_data() local
202 Tensor input = tf_in.make(test_case.sizes, test_case.data_in); in test_runner_hardcode_data()
452 TensorFactory<ScalarType::Float> tf_in; in TEST_F() local
455 tf_in.make(/*sizes=*/{3, 1, 1, 2}, /*data=*/{1, 2, 3, 4, 5, 6}); in TEST_F()
|
| H A D | op__to_dim_order_copy_test.cpp | 96 TensorFactory<INPUT_DTYPE> tf_in; in test_runner_static_cast() local
103 Tensor input = tf_in.make(test_case.sizes, data_in); in test_runner_static_cast()
131 TensorFactory<INPUT_DTYPE> tf_in; in test_runner_to_bool() local
136 Tensor input = tf_in.make({(int)test_case.size()}, data_in); in test_runner_to_bool()
165 TensorFactory<ScalarType::Bool> tf_in; in test_runner_from_bool() local
170 Tensor input = tf_in.make({(int)test_case.size()}, test_case); in test_runner_from_bool()
260 TensorFactory<INPUT_DTYPE> tf_in; in test_runner_hardcode_data() local
275 Tensor input = tf_in.make(test_case.sizes, test_case.data_in); in test_runner_hardcode_data()
479 TensorFactory<ScalarType::Float> tf_in; in TEST_F() local
482 tf_in.make(/*sizes=*/{3, 1, 1, 2}, /*data=*/{1, 2, 3, 4, 5, 6}); in TEST_F()
|
| H A D | op_sum_test.cpp | 41 TensorFactory<IN_DTYPE> tf_in; in test_sum_dim_out_invalid_dimensions() local
45 Tensor self = tf_in.make( in test_sum_dim_out_invalid_dimensions()
79 TensorFactory<IN_DTYPE> tf_in; in test_sum_dim_out_invalid_shape() local
83 Tensor self = tf_in.make( in test_sum_dim_out_invalid_shape()
116 TensorFactory<IN_DTYPE> tf_in; in test_sum_dim_out_dtype() local
119 Tensor self = tf_in.make( in test_sum_dim_out_dtype()
202 self = tf_in.make( in test_sum_dim_out_dtype()
|
| H A D | op_clamp_test.cpp | 473 TensorFactory<ScalarType::Byte> tf_in; in TEST_F() local
478 Tensor in = tf_in.make({1, 1}, {3}); in TEST_F()
489 TensorFactory<ScalarType::Byte> tf_in; in TEST_F() local
494 Tensor in = tf_in.make({}, {5}); in TEST_F()
505 TensorFactory<ScalarType::Short> tf_in; in TEST_F() local
510 Tensor in = tf_in.make({}, {301}); in TEST_F()
521 TensorFactory<ScalarType::Short> tf_in; in TEST_F() local
526 Tensor in = tf_in.make({}, {45}); in TEST_F()
|
| H A D | op_copy_test.cpp | 220 TensorFactory<ScalarType::Int> tf_in; in TEST_F() local
222 Tensor self = tf_in.make(/*sizes=*/{3, 1, 1, 2}, /*data=*/{1, 2, 3, 4, 5, 6}); in TEST_F()
223 Tensor src = tf_in.make(/*sizes=*/{3, 1, 1, 2}, /*data=*/{1, 2, 3, 4, 5, 6}); in TEST_F()
236 TensorFactory<ScalarType::Float> tf_in; in TEST_F() local
238 Tensor self = tf_in.make(/*sizes=*/{3, 1, 1, 2}, /*data=*/{1, 2, 3, 4, 5, 6}); in TEST_F()
239 Tensor src = tf_in.make(/*sizes=*/{3, 1, 1, 2}, /*data=*/{1, 2, 3, 4, 5, 6}); in TEST_F()
|
| H A D | op_var_test.cpp | 42 TensorFactory<IN_DTYPE> tf_in; in test_var_out_invalid_dimensions() local
46 Tensor self = tf_in.make( in test_var_out_invalid_dimensions()
80 TensorFactory<IN_DTYPE> tf_in; in test_var_out_invalid_shape() local
84 Tensor self = tf_in.make( in test_var_out_invalid_shape()
121 TensorFactory<IN_DTYPE> tf_in; in test_var_out_dtype() local
124 Tensor self = tf_in.make( in test_var_out_dtype()
|
| H A D | op_pixel_shuffle_test.cpp | 37 TensorFactory<DTYPE_IN> tf_in; in test_pixel_shuffle() local
43 Tensor out = tf_in.zeros(out_sizes); in test_pixel_shuffle()
46 tf_in.make( in test_pixel_shuffle()
53 tf_in.make( in test_pixel_shuffle()
|
| H A D | op_pixel_unshuffle_test.cpp | 37 TensorFactory<DTYPE_IN> tf_in; in test_pixel_unshuffle() local
43 Tensor out = tf_in.zeros(out_sizes); in test_pixel_unshuffle()
46 tf_in.make( in test_pixel_unshuffle()
52 tf_in.make( in test_pixel_unshuffle()
|
| H A D | op_mean_test.cpp | 41 TensorFactory<IN_DTYPE> tf_in; in test_mean_dim_out_invalid_dimensions() local
45 Tensor self = tf_in.make( in test_mean_dim_out_invalid_dimensions()
77 TensorFactory<IN_DTYPE> tf_in; in test_mean_dim_out_invalid_shape() local
81 Tensor self = tf_in.make( in test_mean_dim_out_invalid_shape()
112 TensorFactory<IN_DTYPE> tf_in; in test_mean_dim_out_dtype() local
115 Tensor self = tf_in.make( in test_mean_dim_out_dtype()
|
| H A D | op_clone_test.cpp | 102 TensorFactory<ScalarType::Int> tf_in; in TEST_F() local
105 tf_in.make(/*sizes=*/{3, 1, 1, 2}, /*data=*/{1, 2, 3, 4, 5, 6}); in TEST_F()
118 TensorFactory<ScalarType::Float> tf_in; in TEST_F() local
121 tf_in.make(/*sizes=*/{3, 1, 1, 2}, /*data=*/{1, 2, 3, 4, 5, 6}); in TEST_F()
|
| H A D | BinaryLogicalOpTest.h | 34 TensorFactory<IN_DTYPE> tf_in; in test_op_out() local
40 using CTYPE1 = typename decltype(tf_in)::ctype; in test_op_out()
58 tf_in.make({1, 4}, test_vector1), in test_op_out()
|
| H A D | op_logical_not_test.cpp | 34 TensorFactory<IN_DTYPE> tf_in; in test_logical_not_out() local
38 Tensor in = tf_in.make( in test_logical_not_out()
44 Tensor bool_in = tf_in.make( in test_logical_not_out()
|
| H A D | op_view_copy_test.cpp | 194 TensorFactory<ScalarType::Int> tf_in; in TEST_F() local
198 Tensor input = tf_in.make(size, /*data=*/{1, 2, 3, 4, 5, 6}); in TEST_F()
217 TensorFactory<ScalarType::Float> tf_in; in TEST_F() local
224 Tensor input = tf_in.make(in_size, /*data=*/{1, 2, 3, 4, 5, 6, 7, 8}); in TEST_F()
|
| H A D | UnaryUfuncRealHBBF16ToFloatHBF16Test.h | 48 TensorFactory<IN_DTYPE> tf_in; variable
53 using IN_CTYPE = typename decltype(tf_in)::ctype;
69 op_out(tf_in.make({1, 6}, test_vector), out);
|
| H A D | op_lift_fresh_copy_test.cpp | 81 TensorFactory<ScalarType::Int> tf_in; in TEST_F() local
83 Tensor self = tf_in.make(/*sizes=*/{3, 1, 1, 2}, /*data=*/{1, 2, 3, 4, 5, 6}); in TEST_F()
|
| H A D | op_any_test.cpp | 65 TensorFactory<IN_DTYPE> tf_in; in test_any_all_out() local
68 Tensor in = tf_in.make( in test_any_all_out()
|
| H A D | op_cumsum_test.cpp | 39 TensorFactory<IN_DTYPE> tf_in; in test_cumsum_out_dtype() local
42 Tensor in = tf_in.make( in test_cumsum_out_dtype()
|
| H A D | op_unsqueeze_copy_test.cpp | 155 TensorFactory<ScalarType::Int> tf_in; in TEST_F() local
157 Tensor input = tf_in.make(/*sizes=*/{3, 1, 2}, /*data=*/{1, 2, 3, 4, 5, 6}); in TEST_F()
|
| H A D | op_repeat_test.cpp | 231 TensorFactory<ScalarType::Int> tf_in; in TEST_F() local
234 Tensor x = tf_in.ones( in TEST_F()
|
| /aosp_15_r20/external/executorch/extension/llm/custom_ops/ |
| H A D | op_tile_crop_test.cpp | 29 TensorFactory<DTYPE_IN> tf_in; in test_tile_crop() local
34 Tensor out = tf_in.zeros(out_sizes); in test_tile_crop()
38 tf_in.make( in test_tile_crop()
47 tf_in.make( in test_tile_crop()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | spacetobatch_op_test.py | 154 tf_in = constant_op.constant(
158 sess.run(array_ops.space_to_batch_nd(tf_in, block_shape, paddings))
164 tf_in = constant_op.constant(
169 sess.run(array_ops.space_to_batch_nd(tf_in, block_shape, paddings))
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| H A D | spacetobatch_op_test.py | 523 tf_in = constant_op.constant(
529 array_ops.space_to_batch_nd(tf_in, block_shape, paddings)
533 tf_in = constant_op.constant(
540 array_ops.space_to_batch_nd(tf_in, block_shape, paddings)
|