Searched refs:extract_scalar_tensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/executorch/runtime/core/exec_aten/util/test/ |
| H A D | tensor_util_test.cpp | 20 using executorch::runtime::extract_scalar_tensor;
275 ok = extract_scalar_tensor(t, &out_##ctype); \ in TEST_F()
288 ok = extract_scalar_tensor(t, &out_##ctype); \ in TEST_F()
301 ok = extract_scalar_tensor(t, &out_##ctype); \ in TEST_F()
313 ok = extract_scalar_tensor(t, &out); in TEST_F()
324 ok = extract_scalar_tensor(t_pos_inf, &value); in TEST_F()
330 ok = extract_scalar_tensor(t_neg_inf, &value); in TEST_F()
336 ok = extract_scalar_tensor(t_nan, &value); in TEST_F()
346 bool ok = extract_scalar_tensor(t, &out); in TEST_F()
354 bool ok = extract_scalar_tensor(t, &out); in TEST_F()
[all …]
|
| /aosp_15_r20/external/executorch/runtime/core/exec_aten/util/ |
| H A D | tensor_util.h | 404 extract_scalar_tensor(scalar_tensor, &out_val), \
1030 bool extract_scalar_tensor(executorch::aten::Tensor tensor, INT_T* out_val) { in extract_scalar_tensor() function
1067 bool extract_scalar_tensor(executorch::aten::Tensor tensor, FLOAT_T* out_val) { in extract_scalar_tensor() function
1105 bool extract_scalar_tensor(executorch::aten::Tensor tensor, BOOL_T* out_val) { in extract_scalar_tensor() function
1341 using ::executorch::runtime::extract_scalar_tensor;
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_fill.cpp | 95 extract_scalar_tensor(b, &b_val); in fill_tensor_out()
|