OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:while_body
(Results
1 - 6
of
6
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/
while_loop_invariant_code_motion_test.cc
64
HloComputation*
while_body
= [&]() {
local
65
HloComputation::Builder builder(TestName() + ".
while_body
");
86
while_body
, init_value));
97
EXPECT_THAT(transformed_while->
while_body
()->instructions(),
106
HloComputation*
while_body
= [&]() {
local
107
HloComputation::Builder builder(TestName() + ".
while_body
");
145
while_body
, init_value));
164
EXPECT_THAT(transformed_while->
while_body
()->instructions(),
168
EXPECT_THAT(transformed_while->
while_body
()->instructions(),
178
HloComputation*
while_body
= [&]()
local
216
HloComputation*
while_body
= [&]() {
local
252
HloComputation*
while_body
= [&]() {
local
291
HloComputation*
while_body
= [&]() {
local
334
HloComputation*
while_body
= [&]() {
local
383
HloComputation*
while_body
;
local
418
HloComputation*
while_body
= [&]() {
local
[
all
...]
while_loop_invariant_code_motion.cc
40
HloComputation*
while_body
= while_instr->
while_body
();
local
50
HloInstruction* while_body_param =
while_body
->parameter_instruction(0);
126
// Populates `gte_set` with the GetTupleElement instructions in `
while_body
`
128
// iterations. Assumes `
while_body
` is the body computation of the while loop
130
static void GatherInvariantGTEs(HloComputation*
while_body
,
133
while_body
->root_instruction()->operands();
138
instr->operand(0) ==
while_body
->parameter_instruction(0) &&
152
//
while_body
(f32[1024, 1024] %param) {
167
HloComputation*
while_body
= while_instr->while_body()
local
[
all
...]
while_loop_simplifier_test.cc
130
auto* true_op = while_op->
while_body
()->AddInstruction(
133
while_op->
while_body
()->root_instruction()));
146
auto*
while_body
= while_op->
while_body
();
local
147
auto* send =
while_body
->AddInstruction(HloInstruction::CreateSend(
148
while_body
->AddInstruction(
151
while_body
->AddInstruction(HloInstruction::CreateSendDone(send));
159
auto*
while_body
= while_op->
while_body
();
local
160
auto* recv =
while_body
->AddInstruction
175
auto*
while_body
= while_op->
while_body
();
local
425
HloComputation*
while_body
= [&]() {
local
[
all
...]
while_loop_simplifier.cc
148
// The
while_body
computation should have the form
155
auto*
while_body
= while_op->
while_body
();
local
156
auto* while_body_root =
while_body
->root_instruction();
164
auto* while_body_param =
while_body
->parameter_instruction(0);
203
// where comp is while_op->
while_body
() or while_op->while_condition().
256
auto*
while_body
= while_op->
while_body
();
local
258
while_body
->root_instruction()->operand(*indvar_tuple_idx);
301
HloComputation*
while_body
= while_op->while_body()
local
[
all
...]
hlo_verifier.cc
286
xla_while->
while_body
()->ComputeProgramShape().result());
772
auto*
while_body
= instruction->
while_body
();
local
776
TF_RET_CHECK(
while_body
->num_parameters() == 1)
778
<<
while_body
->num_parameters() << ": " <<
while_body
->ToString();
795
auto* body_param =
while_body
->parameter_instruction(0)
[
all
...]
hlo_instruction.cc
2012
HloComputation* HloInstruction::
while_body
() const {
function in class:xla::HloInstruction
[
all
...]
Completed in 233 milliseconds