Searched refs:kCollectivePermute (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | thunk.cc | 37 case Thunk::kCollectivePermute: in ThunkKindToString()
|
| D | thunk.h | 48 kCollectivePermute, enumerator
|
| D | collective_permute_thunk.cc | 225 : Thunk(kCollectivePermute, thunk_info), in CollectivePermuteThunk()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_support.cc | 86 case HloOpcode::kCollectivePermute: in EffectiveOperandPrecisionIsOutputPrecision()
|
| D | hlo_opcode.h | 65 V(kCollectivePermute, "collective-permute", 1) \
|
| D | instruction_fusion.cc | 153 case HloOpcode::kCollectivePermute: in IsExpensive()
|
| D | hlo_instruction.cc | 476 case HloOpcode::kCollectivePermute: in CreateFromProto()
489 if (opcode == HloOpcode::kCollectivePermute) { in CreateFromProto()
1078 HloOpcode::kCollectivePermute, shape, operand, source_target_pairs, in CreateCollectivePermute()
1691 case HloOpcode::kCollectivePermute: in CloneWithNewOperands()
2231 case HloOpcode::kCollectivePermute: in IdenticalSlowPath()
3136 case HloOpcode::kCollectivePermute: in Visit()
|
| D | hlo_graph_dumper.cc | 1067 case HloOpcode::kCollectivePermute: in GetInstructionColor()
|
| D | hlo_parser.cc | 166 case HloOpcode::kCollectivePermute: in CanInferShape()
1284 case HloOpcode::kCollectivePermute: in ParseInstructionRhs()
1304 if (opcode == HloOpcode::kCollectivePermute) { in ParseInstructionRhs()
|
| D | sharding_propagation.cc | 182 case HloOpcode::kCollectivePermute: in PickRepresentativeOperand()
|
| D | layout_assignment.cc | 2224 case HloOpcode::kCollectivePermute: in InstructionCanChangeLayout()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 439 return xla::HloOpcode::kCollectivePermute; in MhloToHloOpcode()
|
| D | hlo_function_importer.cc | 313 case HloOpcode::kCollectivePermute: { in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 444 EXPECT_EQ(root->opcode(), HloOpcode::kCollectivePermute); in TEST_F()
|
| D | xla_builder.cc | 3013 return AddInstruction(std::move(instr), HloOpcode::kCollectivePermute, in CollectivePermute()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 274 case HloOpcode::kCollectivePermute: in EmitOp()
|