| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_creation_utils.h | 115 HloComputation* map_computation);
|
| flatten_call_graph_test.cc | 48 HloComputation* map_computation, int64 callsites) {
55 kScalarShape, {last_value}, map_computation));
|
| call_graph_test.cc | 50 HloComputation* map_computation, int64 callsites) {
57 kScalarShape, {last_value}, map_computation));
142 HloComputation* map_computation = local
145 MakeMappingComputation(map_computation, /*callsites=*/5));
159 const CallGraphNode& map_node = call_graph->GetNode(map_computation);
160 EXPECT_EQ(map_computation, map_node.computation());
|
| hlo_creation_utils.cc | 230 HloComputation* map_computation) {
245 operand_shapes, map_computation->ComputeProgramShape(), map_dims));
247 HloInstruction::CreateMap(map_shape, operands, map_computation));
|
| hlo_computation_test.cc | 62 HloComputation* map_computation) {
67 HloInstruction::CreateMap(r0f32_, {param}, map_computation));
86 auto map_computation = local
89 EXPECT_THAT(map_computation->MakeEmbeddedComputationsList(),
|
| buffer_assignment_test.cc | [all...] |
| hlo_instruction_test.cc | 1026 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); local
[all...] |
| hlo_instruction.h | 433 HloComputation* map_computation);
[all...] |
| hlo_instructions.h | 584 HloComputation* map_computation);
[all...] |
| algebraic_simplifier.cc | [all...] |
| hlo_instruction.cc | 776 HloComputation* map_computation)
[all...] |
| hlo_instructions.cc | [all...] |
