Home
last modified time | relevance | path

Searched refs:MakeScalarComputation (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcall_graph_test.cc37 std::unique_ptr<HloComputation> MakeScalarComputation( in MakeScalarComputation() function in xla::__anonbba859f30111::CallGraphTest
99 module->AddEntryComputation(MakeScalarComputation()); in TEST_F()
119 module->AddEntryComputation(MakeScalarComputation()); in TEST_F()
121 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
143 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
174 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
207 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
251 module->AddEmbeddedComputation(MakeScalarComputation(HloOpcode::kCeil)); in TEST_F()
253 module->AddEmbeddedComputation(MakeScalarComputation(HloOpcode::kFloor)); in TEST_F()
315 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
[all …]
Dflatten_call_graph_test.cc36 std::unique_ptr<HloComputation> MakeScalarComputation() { in MakeScalarComputation() function in xla::__anonb6a180e00111::FlattenCallGraphTest
116 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
207 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
230 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()