Home
last modified time | relevance | path

Searched refs:MakeConditionComputation (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dflatten_call_graph_test.cc78 std::unique_ptr<HloComputation> MakeConditionComputation() { in MakeConditionComputation() function in xla::__anonb6a180e00111::FlattenCallGraphTest
114 module->AddEmbeddedComputation(MakeConditionComputation()); in TEST_F()
Dcall_graph_test.cc80 std::unique_ptr<HloComputation> MakeConditionComputation() { in MakeConditionComputation() function in xla::__anonbba859f30111::CallGraphTest
313 module->AddEmbeddedComputation(MakeConditionComputation()); in TEST_F()
444 module->AddEmbeddedComputation(MakeConditionComputation()); in TEST_F()
Dhlo_rematerialization_test.cc130 std::unique_ptr<HloComputation> MakeConditionComputation() { in MakeConditionComputation() function in xla::__anon7066800c0111::HloRematerializationTest