Home
last modified time | relevance | path

Searched refs:c_computation (Results 1 – 4 of 4) 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.cc1440 auto c_computation = c_builder.Build(); in TEST_F() local
1446 HloInstruction::CreateCall(tuple_shape, {b_param}, c_computation.get())); in TEST_F()
1461 module->AddEmbeddedComputation(std::move(c_computation)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py142 def compile(self, c_computation, argument_shapes, result_shape, argument
144 return c_computation.Compile(argument_shapes, compile_options, self.client)
184 def compile(self, c_computation, argument_shapes, result_shape, argument
187 c_computation.GetSerializedProto(), argument_shapes, result_shape,
713 def __init__(self, c_computation, backend=None): argument
714 self._c_computation = c_computation