| /aosp_15_r20/external/executorch/extension/llm/custom_ops/spinquant/test/ |
| H A D | op_fast_hadamard_transform_test.cpp | 33 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST() local
41 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST() local
51 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST() local
67 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST() local
88 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST() local
111 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST() local
|
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_native_batch_norm_test.cpp | 109 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
195 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
355 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
674 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
827 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
893 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
981 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
1021 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
1054 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
|
| H A D | op_diagonal_copy_test.cpp | 45 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
55 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
66 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
|
| H A D | op_replication_pad2d_test.cpp | 43 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
74 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
86 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
|
| H A D | op_reflection_pad2d_test.cpp | 43 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
74 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
86 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
|
| H A D | op_reflection_pad1d_test.cpp | 43 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
56 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
68 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
|
| H A D | op_reflection_pad3d_test.cpp | 43 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
82 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
95 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
|
| H A D | op_replication_pad1d_test.cpp | 43 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
56 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
68 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
|
| H A D | op_replication_pad3d_test.cpp | 43 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
82 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
95 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
|
| H A D | op_prod_test.cpp | 60 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
71 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
84 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
|
| H A D | op_avg_pool2d_test.cpp | 45 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
197 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
350 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
|
| H A D | op_max_pool2d_with_indices_test.cpp | 46 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
113 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
216 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
|
| H A D | op_flip_test.cpp | 39 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
53 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
|
| H A D | op_squeeze_copy_test.cpp | 325 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
360 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
385 torch::executor::testing::TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST_F() local
|
| H A D | op_atan2_test.cpp | 32 TensorFactory<ScalarType::Float> tfFloat; in TEST() local
|
| H A D | op_maximum_test.cpp | 30 TensorFactory<ScalarType::Float> tfFloat; in TEST() local
|
| H A D | op_roll_test.cpp | 43 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
|
| H A D | op_pdist_forward_test.cpp | 39 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
|
| H A D | op_split_with_sizes_copy_test.cpp | 33 tfFloat; in test_tensor_shape_dynamism() local
|
| H A D | op_cdist_forward_test.cpp | 46 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
|
| H A D | op_native_group_norm_test.cpp | 41 TensorFactory<ScalarType::Float> tfFloat; in TEST() local
|
| H A D | op_topk_test.cpp | 122 TensorFactory<ScalarType::Float> tfFloat; in TEST_F() local
|
| /aosp_15_r20/external/executorch/extension/llm/custom_ops/ |
| H A D | op_sdpa_test.cpp | 41 TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST() local
127 TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST() local
156 TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST() local
185 TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST() local
261 TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST() local
483 TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST() local
|
| H A D | op_sdpa_with_kv_cache_test.cpp | 83 TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST() local
364 TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST() local
528 TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST() local
811 TensorFactory<exec_aten::ScalarType::Float> tfFloat; in TEST() local
|