HomeSort by relevance Sort by last modified time
    Searched defs:subcomputation (Results 1 - 3 of 3) 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 200 HloComputation* subcomputation = local
207 HloInstruction::CreateCall(kScalarShape, {param0}, subcomputation));
209 HloInstruction::CreateMap(kScalarShape, {call}, subcomputation));
225 UnorderedElementsAre(subcomputation));
232 UnorderedElementsAre(subcomputation));
236 const CallGraphNode& sub_node = call_graph->GetNode(subcomputation);
hlo_rematerialization_test.cc 403 // F32[1024] %call_1 = call(Subcomputation, {%add_1})
405 // F32[1024] %call_2 = call(SubComputation, {%add_2})
407 // F32[1024] %call_3 = call(Subcomputation, {%add_3})
410 // Subcomputation:
415 // The value %bcast is live across each call of Subcomputation (which requires
420 HloComputation* subcomputation = nullptr; local
422 auto builder = HloComputation::Builder(TestName() + ".subcomputation");
431 subcomputation = module->AddEmbeddedComputation(builder.Build());
442 HloInstruction::CreateCall(vec1024_shape_, {add_1}, subcomputation));
446 HloInstruction::CreateCall(vec1024_shape_, {add_2}, subcomputation));
536 HloComputation* subcomputation = nullptr; local
    [all...]

Completed in 601 milliseconds