Searched refs:token_operand (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.h | 537 const Shape& infeed_shape, HloInstruction* token_operand,
545 HloInstruction* token_operand, absl::string_view outfeed_config);
807 HloInstruction* data_operand, HloInstruction* token_operand);
|
| D | hlo_instruction.cc | 863 const Shape& infeed_shape, HloInstruction* token_operand, in CreateInfeed() argument
865 return absl::make_unique<HloInfeedInstruction>(infeed_shape, token_operand, in CreateInfeed()
871 HloInstruction* token_operand, absl::string_view outfeed_config) { in CreateOutfeed() argument
873 outfeed_shape, operand, token_operand, outfeed_config); in CreateOutfeed()
932 HloInstruction* token_operand) { in CreateAddDependency() argument
936 instruction->AppendOperand(token_operand); in CreateAddDependency()
|
| D | hlo_instructions.cc | 1773 HloInstruction* token_operand, in HloInfeedInstruction() argument
1779 AppendOperand(token_operand);
1814 HloInstruction* token_operand, in HloOutfeedInstruction() argument
1820 AppendOperand(token_operand); in HloOutfeedInstruction()
|
| D | hlo_instructions.h | 965 HloInstruction* token_operand,
1002 HloInstruction* token_operand,
|