OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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