Searched refs:HloValuesAt (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_dataflow_analysis_test.cc | 60 std::vector<HloValue> HloValuesAt(const HloInstruction* instruction, in HloValuesAt() function in xla::__anondaadebfd0111::HloDataflowAnalysisTest
389 EXPECT_THAT(HloValuesAt(subparam0), in TEST_P()
392 EXPECT_THAT(HloValuesAt(subparam1), in TEST_P()
717 EXPECT_THAT(HloValuesAt(inner_param, /*index=*/{0}), in TEST_P()
726 EXPECT_THAT(HloValuesAt(inner_param, /*index=*/{0}), in TEST_P()
741 EXPECT_THAT(HloValuesAt(inner_param, /*index=*/{1}), in TEST_P()
745 EXPECT_THAT(HloValuesAt(nested_while, /*index=*/{0}), in TEST_P()
747 EXPECT_THAT(HloValuesAt(nested_while, /*index=*/{1}), in TEST_P()
751 EXPECT_THAT(HloValuesAt(entry_while, /*index=*/{0}), in TEST_P()
754 EXPECT_THAT(HloValuesAt(entry_while, /*index=*/{1}), in TEST_P()
[all …]
|