HomeSort by relevance Sort by last modified time
    Searched refs:data_init (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
while_transformer_test.cc 93 auto data_init = builder.AddInstruction( local
99 HloInstruction::CreateTuple({induction_var_init, data_init}))
101 HloInstruction::CreateTuple({data_init, induction_var_init}));
  /external/tensorflow/tensorflow/compiler/xla/service/
copy_insertion_test.cc 641 auto data_init = builder.AddInstruction( local
647 HloInstruction::CreateTuple({data_init, data_init}));
657 HloInstruction::CreateTuple({induction_var_init, data_init}));
666 auto data_init = builder.AddInstruction( local
669 return BuildWhileInstructionWithCustomInit(loop_state_shape_, data_init,
675 auto data_init = builder.AddInstruction( local
676 HloInstruction::CreateParameter(0, data_shape_, "data_init"));
677 return BuildWhileInstructionWithCustomInit(loop_state_shape_, data_init,
698 auto data_init = builder.AddInstruction(HloInstruction::CreateTernary local
712 auto data_init = local
723 auto data_init = builder.AddInstruction( local
    [all...]

Completed in 417 milliseconds