Home
last modified time | relevance | path

Searched defs:expected_tensors (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmodel_pruner_test.cc88 auto expected_tensors = EvaluateNodes(item.graph, fetch); in TEST_F() local
130 auto expected_tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
178 auto expected_tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
225 auto expected_tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
267 auto expected_tensors = EvaluateNodes(item.graph, fetch); in TEST_F() local
308 auto expected_tensors = EvaluateNodes( in TEST_F() local
354 auto expected_tensors = EvaluateNodes(item.graph, fetch, {{"a", a_t}}); in TEST_F() local
441 auto expected_tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
496 auto expected_tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
H A Dconstant_folding_test.cc1125 auto expected_tensors = in TEST_F() local
1194 auto expected_tensors = in TEST_F() local
4389 std::vector<Tensor> expected_tensors = EvaluateNodes(item.graph, item.fetch); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/io_ops/
H A Dparse_single_example_op_test.py65 def _compare_output_to_expected(tester, dict_tensors, expected_tensors, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
H A Dparse_example_dataset_test.py56 def _compare_output_to_expected(self, dict_tensors, expected_tensors): argument
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dnn_utils.cpp862 std::vector<torch::Tensor> expected_tensors; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_test_base.cc240 std::vector<Tensor> expected_tensors, in ExpectEqual()
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Ddistributed_test.py3614 self, output_tensor_lists, input_tensors, expected_tensors, group_id argument