Home
last modified time | relevance | path

Searched defs:tf_index (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/executorch/kernels/test/
H A Dop_scatter_test.cpp40 TensorFactory<ScalarType::Long> tf_index; in test_scatter_src_out() local
155 TensorFactory<ScalarType::Long> tf_index; in test_scatter_src_out_invalid_dim() local
223 TensorFactory<ScalarType::Long> tf_index; in test_scatter_value_out() local
311 TensorFactory<ScalarType::Long> tf_index; in test_scatter_value_out_invalid_dim() local
356 TensorFactory<ScalarType::Long> tf_index; in test_dynamic_shape() local
389 TensorFactory<ScalarType::Long> tf_index; in TEST_F() local
465 TensorFactory<ScalarType::Long> tf_index; in TEST_F() local
477 TensorFactory<ScalarType::Long> tf_index; in TEST_F() local
489 TensorFactory<ScalarType::Long> tf_index; in TEST_F() local
501 TensorFactory<ScalarType::Long> tf_index; in TEST_F() local
[all …]
H A Dop_gather_test.cpp40 TensorFactory<ScalarType::Long> tf_index; in test_gather_out() local
137 TensorFactory<ScalarType::Long> tf_index; in test_gather_out_invalid_dim() local
182 TensorFactory<ScalarType::Long> tf_index; in test_dynamic_shape() local
202 TensorFactory<ScalarType::Long> tf_index; in TEST_F() local
279 TensorFactory<ScalarType::Long> tf_index; in TEST_F() local
292 TensorFactory<ScalarType::Long> tf_index; in TEST_F() local
304 TensorFactory<ScalarType::Long> tf_index; in TEST_F() local
317 TensorFactory<ScalarType::Long> tf_index; in TEST_F() local
330 TensorFactory<ScalarType::Long> tf_index; in TEST_F() local
343 TensorFactory<ScalarType::Long> tf_index; in TEST_F() local
[all …]
H A Dop_scatter_add_test.cpp40 TensorFactory<ScalarType::Long> tf_index; in test_scatter_add_out() local
155 TensorFactory<ScalarType::Long> tf_index; in test_scatter_add_out_invalid_dim() local
210 TensorFactory<ScalarType::Long> tf_index; in test_scatter_add_out_mismatched_shape() local
263 TensorFactory<ScalarType::Long> tf_index; in test_dynamic_shape() local
289 TensorFactory<ScalarType::Long> tf_index; in TEST_F() local
H A Dop_index_select_test.cpp90 TensorFactory<ScalarType::Long> tf_index; in test_dynamic_shape() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc3867 const int tf_index = tf_pad_index[i]; in ConvertPad() local