Home
last modified time | relevance | path

Searched refs:cond_computation (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dflatten_call_graph_test.cc113 HloComputation* cond_computation = in TEST_F() local
128 kScalarShape, cond_computation, b_computation, call)); in TEST_F()
138 kScalarShape, cond_computation, a_computation, param0)); in TEST_F()
154 HloComputation* cond_computation; in TEST_F() local
165 cond_computation = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
174 ShapeUtil::MakeShape(PRED, {}), cond_computation, cond_computation, in TEST_F()
181 const CallGraphNode& cond_node = call_graph->GetNode(cond_computation); in TEST_F()
189 const CallGraphNode& cond_node = call_graph->GetNode(cond_computation); in TEST_F()
Dcall_graph_test.cc312 HloComputation* cond_computation = in TEST_F() local
327 kScalarShape, cond_computation, b_computation, call)); in TEST_F()
337 kScalarShape, cond_computation, a_computation, param0)); in TEST_F()
349 const CallGraphNode& cond_node = call_graph->GetNode(cond_computation); in TEST_F()
364 UnorderedElementsAre(cond_computation, a_computation)); in TEST_F()
393 EXPECT_LT(index_of(cond_computation), index_of(a_computation)); in TEST_F()
404 EXPECT_TRUE(call_graph->Dominates(entry_computation, cond_computation)); in TEST_F()
408 EXPECT_FALSE(call_graph->Dominates(cond_computation, entry_computation)); in TEST_F()
416 EXPECT_FALSE(call_graph->Dominates(a_computation, cond_computation)); in TEST_F()
420 EXPECT_FALSE(call_graph->Dominates(b_computation, cond_computation)); in TEST_F()
[all …]
Dwhile_util.cc158 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloComputation> cond_computation, in MakeCountedLoopConditionComputation()
162 HloInstruction* trip_count_constant = cond_computation->AddInstruction( in MakeCountedLoopConditionComputation()
165 HloInstruction* param = cond_computation->parameter_instruction(0); in MakeCountedLoopConditionComputation()
172 cond_computation->set_root_instruction(compare); in MakeCountedLoopConditionComputation()
173 return std::move(cond_computation); in MakeCountedLoopConditionComputation()
Dhlo_live_range_test.cc279 HloComputation* cond_computation = in TEST_F() local
319 tuple_shape, cond_computation, body_computation, tuple)); in TEST_F()
324 schedule.set_sequence(cond_computation, in TEST_F()
Dheap_simulator_test.cc60 HloComputation* cond_computation = in TEST_F() local
82 tuple_shape, cond_computation, body_computation, tuple)); in TEST_F()
91 schedule.set_sequence(cond_computation, in TEST_F()
151 auto cond_computation = module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F() local
174 r1f32, cond_computation, body_computation, while_init)); in TEST_F()
200 schedule.set_sequence(cond_computation, cond_vec); in TEST_F()
208 memory_by_computation[cond_computation] = 5; in TEST_F()
902 HloComputation* cond_computation = in TEST_F() local
915 tuple_shape, cond_computation, body_computation, param)); in TEST_F()
Dhlo_dce_test.cc268 auto cond_computation = module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F() local
290 shape, cond_computation, body_computation, param)); in TEST_F()
Dmemory_space_assignment_test.cc756 HloComputation* cond_computation = in TEST_P() local
797 tuple_shape, cond_computation, body_computation, tuple)); in TEST_P()
802 schedule.set_sequence(cond_computation, in TEST_P()
2639 HloComputation* cond_computation = in TEST_P() local
2682 tuple_shape, cond_computation, body_computation, tuple)); in TEST_P()
2691 schedule.set_sequence(cond_computation, in TEST_P()
2961 HloComputation* cond_computation = in TEST_P() local
3011 tuple_shape, cond_computation, body_computation, tuple)); in TEST_P()
3020 schedule.set_sequence(cond_computation, in TEST_P()
3052 HloComputation* cond_computation = in TEST_P() local
[all …]
Dbuffer_assignment_test.cc792 HloComputation* cond_computation = in TEST_F() local
828 t_s32_f32v10_color1, cond_computation, body_computation, tuple)); in TEST_F()
Dhlo_dataflow_analysis_test.cc2742 HloComputation* cond_computation = in TEST_F() local
2751 data_shape, cond_computation, body_computation, data)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dwhile_test.cc968 TF_ASSERT_OK_AND_ASSIGN(auto cond_computation, cond.Build()); in XLA_TEST_F()
970 While(cond_computation, body_computation, t); in XLA_TEST_F()
996 TF_ASSERT_OK_AND_ASSIGN(auto cond_computation, cond.Build()); in XLA_TEST_F()
998 While(cond_computation, body_computation, p); in XLA_TEST_F()
1022 TF_ASSERT_OK_AND_ASSIGN(auto cond_computation, cond.Build()); in XLA_TEST_F()
1024 While(cond_computation, body_computation, p); in XLA_TEST_F()
1061 TF_ASSERT_OK_AND_ASSIGN(auto cond_computation, cond.Build()); in XLA_TEST_F()
1063 While(cond_computation, body_computation, p); in XLA_TEST_F()