| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | fusion.mlir | 19 }) { fusion_kind = "kLoop" } : (tensor<f32>, tensor<f32>) -> tensor<f32>
25 }) { fusion_kind="kLoop" } : (tensor<f32>, tensor<f32>) -> (tensor<f32>, tensor<f32>)
|
| D | fusion.hlotxt | 8 // CHECK: }) {fusion_kind = "kLoop"} : (tensor<f32>, tensor<f32>) -> tensor<f32>
11 // CHECK: }) {fusion_kind = "kLoop"} : (tensor<f32>, tensor<f32>) -> tuple<tensor<f32>, tensor<f3…
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | multi_output_fusion.cc | 62 instr1->fusion_kind() != instr2->fusion_kind()) || in LegalToFuse()
|
| D | gpu_fusible.cc | 197 instr.fusion_kind() == HloInstruction::FusionKind::kInput && in IsInputFusibleScatter()
225 instr.fusion_kind() == HloInstruction::FusionKind::kLoop) || in IsLoopFusible()
|
| D | instruction_fusion_test.cc | 388 EXPECT_EQ(root->fusion_kind(), HloInstruction::FusionKind::kLoop); in TEST_F()
529 EXPECT_EQ(root->operand(0)->fusion_kind(), in TEST_F()
|
| D | ir_emitter.cc | 579 CHECK_EQ(HloInstruction::FusionKind::kLoop, fusion->fusion_kind()); in HandleFusion()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.cc | 352 TF_RET_CHECK(!proto.fusion_kind().empty()); in CreateFromProto()
353 TF_ASSIGN_OR_RETURN(FusionKind fusion_kind, in CreateFromProto()
354 StringToFusionKind(proto.fusion_kind())); in CreateFromProto()
366 CreateFusion(shape, fusion_kind, all_operands(), fused_computation); in CreateFromProto()
1485 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root) { in CreateFusion() argument
1486 return absl::make_unique<HloFusionInstruction>(shape, fusion_kind, in CreateFusion()
1491 const Shape& shape, FusionKind fusion_kind, in CreateFusion() argument
1494 return absl::make_unique<HloFusionInstruction>(shape, fusion_kind, operands, in CreateFusion()
3003 return opcode() == HloOpcode::kFusion && fusion_kind() == FusionKind::kInput; in IsInputFusion()
3007 return opcode() == HloOpcode::kFusion && fusion_kind() == FusionKind::kLoop; in IsLoopFusion()
[all …]
|
| D | hlo_instructions.cc | 1367 FusionKind fusion_kind, in HloFusionInstruction() argument
1369 : HloInstruction(HloOpcode::kFusion, shape), fusion_kind_(fusion_kind) { in HloFusionInstruction()
1378 const Shape& shape, FusionKind fusion_kind, in HloFusionInstruction() argument
1381 : HloInstruction(HloOpcode::kFusion, shape), fusion_kind_(fusion_kind) { in HloFusionInstruction()
1391 switch (fusion_kind()) { in ToCategory()
1405 proto.set_fusion_kind(xla::ToString(fusion_kind())); in ToProto()
1809 return {StrCat("kind=", xla::ToString(fusion_kind()))}; in ExtraAttributesToStringImpl()
1816 return fusion_kind() == other.fusion_kind() && in IdenticalSlowPath()
1839 shape, fusion_kind(), new_operands, new_fused_computation); in CloneWithNewOperandsImpl()
|
| D | hlo_instruction.h | 968 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root);
971 const Shape& shape, FusionKind fusion_kind,
1760 FusionKind fusion_kind() const;
|
| D | hlo_computation.h | 284 HloInstruction::FusionKind fusion_kind);
|
| D | hlo_computation.cc | 744 HloInstruction::FusionKind fusion_kind) { in CreateFusionInstruction() argument
747 HloInstruction::CreateFusion(root->shape(), fusion_kind, root)); in CreateFusionInstruction()
|
| D | instruction_fusion.cc | 626 if (kind != fusion_instruction->fusion_kind()) { in AddFusionInstruction()
|
| D | hlo_instructions.h | 891 explicit HloFusionInstruction(const Shape& shape, FusionKind fusion_kind,
894 explicit HloFusionInstruction(const Shape& shape, FusionKind fusion_kind,
973 FusionKind fusion_kind() const { return fusion_kind_; } in fusion_kind() function
|
| D | bfloat16_propagation.cc | 354 inst->fusion_kind() == HloInstruction::FusionKind::kCustom) { in ShouldKeepPrecisionUnchanged()
|
| D | hlo.proto | 72 string fusion_kind = 11; field
|
| D | hlo_graph_dumper.cc | 1114 : StrCat(":", xla::ToString(instr->fusion_kind()))); in GetInstructionNodeLabel()
|
| D | hlo_parser.cc | 2075 optional<HloInstruction::FusionKind> fusion_kind; in ParseInstructionRhs() local
2076 attrs["kind"] = {/*required=*/true, AttrTy::kFusionKind, &fusion_kind}; in ParseInstructionRhs()
2081 shape, *fusion_kind, operands, *fusion_computation)); in ParseInstructionRhs()
|
| D | pattern_matcher.h | 1525 if (inst->fusion_kind() != kind_) {
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 266 HloInstruction::FusionKind fusion_kind = in RunFusionAndCheckOpcodesWereFused() argument
275 EXPECT_EQ(root->fusion_kind(), fusion_kind); in RunFusionAndCheckOpcodesWereFused()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | interactive_graphviz.cc | 342 std::cout << " Fusion kind: " << xla::ToString(instr->fusion_kind()) in DoInfoCommand()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 1063 if (!op.fusion_kind()) { in ExportXlaOp()
1079 absl::string_view(op.fusion_kind()->data(), op.fusion_kind()->size()), in ExportXlaOp()
|
| D | hlo_function_importer.cc | 803 builder_->getStringAttr(xla::ToString(instruction->fusion_kind()))); in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | ops.mlir | 715 }) { fusion_kind = "kLoop" } : (tensor<f32>, tensor<f32>) -> tensor<f32>
741 …}) { fusion_kind = "kLoop" } : (tuple<tuple<tensor<f32>>, tensor<f32>>, tuple<tensor<f32>>) -> tup…
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops.td | 1375 OptionalAttr<HLO_FusionKindAttr>:$fusion_kind
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 156 absl::string_view fusion_kind, in BuildFusion() argument
160 instr.set_fusion_kind(std::string(fusion_kind)); in BuildFusion()
|