HomeSort by relevance Sort by last modified time
    Searched defs:subcomputation (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_copy_insertion_test.cc 113 auto sub_builder = HloComputation::Builder("subcomputation");
122 HloComputation* subcomputation = local
126 HloInstruction::CreateCall(tuple_shape, {param}, subcomputation));
132 EXPECT_EQ(CountCopies(*subcomputation), 2);
133 EXPECT_THAT(subcomputation->root_instruction(),
  /external/tensorflow/tensorflow/compiler/xla/service/
call_graph_test.cc 206 HloComputation* subcomputation = local
213 HloInstruction::CreateCall(kScalarShape, {param0}, subcomputation));
215 HloInstruction::CreateMap(kScalarShape, {call}, subcomputation));
231 UnorderedElementsAre(subcomputation));
238 UnorderedElementsAre(subcomputation));
242 const CallGraphNode& sub_node = call_graph->GetNode(subcomputation);
hlo_ordering_test.cc 448 // subcomputation:
455 // ROOT %call = call({%c}), subcomputation
468 HloComputation* subcomputation = module->AddEmbeddedComputation( local
475 HloInstruction::CreateCall(scalar_shape, {c}, subcomputation));
479 schedule.set_sequence(subcomputation, {param, root, dead});
hlo_rematerialization_test.cc 408 // F32[1024] %call_1 = call(Subcomputation, {%add_1})
410 // F32[1024] %call_2 = call(SubComputation, {%add_2})
412 // F32[1024] %call_3 = call(Subcomputation, {%add_3})
415 // Subcomputation:
420 // The value %bcast is live across each call of Subcomputation (which requires
425 HloComputation* subcomputation = nullptr; local
427 auto builder = HloComputation::Builder(TestName() + ".subcomputation");
436 subcomputation = module->AddEmbeddedComputation(builder.Build());
447 HloInstruction::CreateCall(vec1024_shape_, {add_1}, subcomputation));
451 HloInstruction::CreateCall(vec1024_shape_, {add_2}, subcomputation));
584 HloComputation* subcomputation = nullptr; local
    [all...]

Completed in 693 milliseconds