Home
last modified time | relevance | path

Searched refs:c_computation (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcall_graph_test.cc314 HloComputation* c_computation = in TEST_F() local
317 MakeMappingComputation(c_computation, /*callsites=*/1)); in TEST_F()
325 HloInstruction::CreateCall(kScalarShape, {param0}, c_computation)); in TEST_F()
348 const CallGraphNode& c_node = call_graph->GetNode(c_computation); in TEST_F()
394 EXPECT_LT(index_of(c_computation), index_of(b_computation)); in TEST_F()
403 EXPECT_TRUE(call_graph->Dominates(entry_computation, c_computation)); in TEST_F()
407 EXPECT_FALSE(call_graph->Dominates(c_computation, entry_computation)); in TEST_F()
413 EXPECT_TRUE(call_graph->Dominates(a_computation, c_computation)); in TEST_F()
415 EXPECT_FALSE(call_graph->Dominates(c_computation, a_computation)); in TEST_F()
419 EXPECT_FALSE(call_graph->Dominates(b_computation, c_computation)); in TEST_F()
[all …]
Dflatten_call_graph_test.cc115 HloComputation* c_computation = in TEST_F() local
118 MakeMappingComputation(c_computation, /*callsites=*/1)); in TEST_F()
126 HloInstruction::CreateCall(kScalarShape, {param0}, c_computation)); in TEST_F()
146 const CallGraphNode& c_node = flat_call_graph->GetNode(c_computation); in TEST_F()
206 HloComputation* c_computation = in TEST_F() local
210 MakeCallingComputation(c_computation, /*callsites=*/2, ".B")); in TEST_F()
220 const CallGraphNode& c_node = call_graph->GetNode(c_computation); in TEST_F()
Dbuffer_assignment_test.cc1619 auto c_computation = c_builder.Build(); in TEST_F() local
1625 HloInstruction::CreateCall(tuple_shape, {b_param}, c_computation.get())); in TEST_F()
1640 module->AddEmbeddedComputation(std::move(c_computation)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.py92 def compile(self, c_computation, compile_options): argument
103 return _tpu_client.TpuExecutable.Compile(c_computation,
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py141 def compile(self, c_computation, compile_options): argument
152 return _xla.LocalExecutable.Compile(c_computation,
536 def __init__(self, c_computation, backend=None): argument
537 self._c_computation = c_computation