Home
last modified time | relevance | path

Searched refs:test_tensors (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/executorch/backends/arm/test/ops/
H A Dtest_slice.py23 test_tensors = [(torch.ones(n),) for n in sizes] variable in TestSimpleSlice.Slice
111 @parameterized.expand(Slice.test_tensors)
115 @parameterized.expand(Slice.test_tensors[:2])
119 @parameterized.expand(Slice.test_tensors[2:])
123 @parameterized.expand(Slice.test_tensors)
127 @parameterized.expand(Slice.test_tensors)
/aosp_15_r20/external/pytorch/test/
H A Dtest_type_promotion.py847 test_tensors = [expected, dense1, sparse1, dense2, sparse2]
848 e, d1, s1, d2, s2 = [x.clone() for x in test_tensors] if inplace else test_tensors
862 e, d1, s1, d2, s2 = (x.clone() for x in test_tensors)
883 e, d1, s1, d2, s2 = (x.clone() for x in test_tensors)
/aosp_15_r20/external/pytorch/test/ao/sparsity/
H A Dtest_data_sparsifier.py461 def test_tensors(self): member in TestBaseDataSparsifier
535 def test_tensors(self): member in TestNormDataSparsifiers
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_quantized_api_test.cpp369 std::array<at::Tensor, 4> test_tensors = { in TEST_F() local
381 for (auto in_cpu : test_tensors) { in TEST_F()
402 std::array<at::Tensor, 1> test_tensors = { in TEST_F() local
407 for (auto in_cpu : test_tensors) { in TEST_F()
H A Dvulkan_api_test.cpp338 at::Tensor test_tensors[] = { in TEST_F() local
349 for (auto in_cpu : test_tensors) { in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcontrol_flow_ops_test.py697 def test_tensors(self): member in DataTypesTest