| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_dce_test.cc | 243 auto body_computation = module->AddEmbeddedComputation(body_builder.Build()); local
250 shape, cond_computation, body_computation, param));
|
| hlo_memory_scheduler_test.cc | 272 auto body_computation = module->AddEmbeddedComputation(body_builder.Build()); local
280 r1f32, cond_computation, body_computation, while_init));
297 memory_by_computation[body_computation] = 16;
|
| hlo_rematerialization_test.cc | 238 HloComputation* body_computation = module->AddEmbeddedComputation( local
242 while_cond, /*while_body=*/body_computation));
245 EXPECT_EQ(body_computation->instruction_count(), 8);
257 EXPECT_EQ(body_computation->instruction_count(), 8);
274 HloComputation* body_computation = module->AddEmbeddedComputation( local
278 while_cond, /*while_body=*/body_computation));
281 EXPECT_EQ(body_computation->instruction_count(), 8);
290 EXPECT_EQ(body_computation->instruction_count(), 9);
|
| heap_simulator_test.cc | 66 HloComputation* body_computation = local
81 tuple_shape, cond_computation, body_computation, tuple));
92 schedule.set_sequence(body_computation, {body_param});
163 auto body_computation = module->AddEmbeddedComputation(body_builder.Build()); local
173 r1f32, cond_computation, body_computation, while_init));
200 schedule.set_sequence(body_computation, while_body_vec);
208 memory_by_computation[body_computation] = 16;
715 HloComputation* body_computation = local
722 tuple_shape, cond_computation, body_computation, param));
[all...] |
| tuple_points_to_analysis_test.cc | 1205 HloComputation* body_computation = local
[all...] |
| hlo_dataflow_analysis_test.cc | 2611 HloComputation* body_computation = local
[all...] |
