Searched refs:ref_output_tensors (Results 1 – 1 of 1) sorted by relevance
34 const auto& ref_output_tensors = in test_aoti() local50 ASSERT_TRUE(torch::allclose(ref_output_tensors[0], actual_output_tensors[0])); in test_aoti()71 const auto& ref_output_tensors = in test_aoti_script() local74 ASSERT_EQ(outputs.size(), ref_output_tensors.size()); in test_aoti_script()75 for (size_t i = 0; i < ref_output_tensors.size(); i++) { in test_aoti_script()76 ASSERT_TRUE(torch::allclose(outputs[i].toTensor(), ref_output_tensors[i])); in test_aoti_script()101 const auto& ref_output_tensors = in test_aoti_constants_update() local129 ASSERT_TRUE(torch::allclose(ref_output_tensors[0], actual_output_tensors[0])); in test_aoti_constants_update()143 torch::allclose(ref_output_tensors[0], actual_output_tensors[0])); in test_aoti_constants_update()148 torch::allclose(ref_output_tensors[0], actual_output_tensors[0])); in test_aoti_constants_update()[all …]