| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| instruction_fusion.cc | 79 HloInstruction::FusionKind::kInput == consumer->fusion_kind())) &&
102 return consumer->fusion_kind();
|
| fusion_merger.cc | 210 if (fusion->fusion_kind() != HloInstruction::FusionKind::kLoop) {
226 instruction->fusion_kind() ==
|
| ir_emission_utils.cc | 85 hlo.fusion_kind() == HloInstruction::FusionKind::kTransposeDot &&
|
| ir_emitter.cc | 686 CHECK(HloInstruction::FusionKind::kLoop == fusion->fusion_kind());
[all...] |
| ir_emitter_unnested.cc | 497 if (HloInstruction::FusionKind::kInput == fusion->fusion_kind()) {
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| liveness_util.cc | 42 user->fusion_kind() == HloInstruction::FusionKind::kLoop) {
79 user->fusion_kind() == HloInstruction::FusionKind::kLoop) {
198 if (user->fusion_kind() == HloInstruction::FusionKind::kLoop &&
208 } else if (user->fusion_kind() == HloInstruction::FusionKind::kOutput &&
221 operand->fusion_kind() ==
303 if (user->fusion_kind() == HloInstruction::FusionKind::kLoop &&
313 } else if (user->fusion_kind() == HloInstruction::FusionKind::kOutput &&
326 operand->fusion_kind() ==
|
| reduce_precision_insertion.cc | 125 (instruction->fusion_kind() == HloInstruction::FusionKind::kLoop ||
126 instruction->fusion_kind() == HloInstruction::FusionKind::kInput)) {
176 (instruction->fusion_kind() == HloInstruction::FusionKind::kLoop ||
177 instruction->fusion_kind() == HloInstruction::FusionKind::kOutput)) {
|
| instruction_fusion.cc | 408 if (kind != fusion_instruction->fusion_kind()) {
431 consumer->fusion_kind() != HloInstruction::FusionKind::kLoop &&
432 consumer->fusion_kind() != HloInstruction::FusionKind::kInput &&
433 consumer->fusion_kind() != HloInstruction::FusionKind::kOutput) {
|
| hlo.proto | 60 string fusion_kind = 11;
|
| hlo_instruction.cc | 80 TF_RET_CHECK(!proto.fusion_kind().empty());
82 StringToFusionKind(proto.fusion_kind()));
769 static string FusionNodeName(HloInstruction::FusionKind fusion_kind) {
770 switch (fusion_kind) {
779 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root) {
781 instruction->fusion_kind_ = fusion_kind;
782 instruction->name_ = FusionNodeName(fusion_kind);
[all...] |
| hlo_computation.h | 225 // `fusion_kind` indicates the type of the fusion, e.g., loop fusion or fusion
233 HloInstruction::FusionKind fusion_kind);
|
| hlo_tfgraph_builder.cc | 44 string fusion_name = ToString(instruction->fusion_kind());
|
| hlo_instruction.h | 465 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root);
468 const Shape& shape, FusionKind fusion_kind,
909 FusionKind fusion_kind() const { function in class:xla::HloInstruction
[all...] |
| hlo_computation.cc | 456 HloInstruction::FusionKind fusion_kind) {
459 HloInstruction::CreateFusion(root->shape(), fusion_kind, root));
|
| hlo_graph_dumper.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| ops.h | 45 fusion->fusion_kind() != HloInstruction::FusionKind::kLoop) {
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_instruction_fusion.cc | 137 consumer->fusion_kind() == HloInstruction::FusionKind::kLoop) {
|
| cpu_instruction_fusion_test.cc | 186 EXPECT_EQ(computation->root_instruction()->fusion_kind(),
190 EXPECT_EQ(computation->root_instruction()->fusion_kind(),
200 HloInstruction::FusionKind fusion_kind =
209 EXPECT_EQ(root->fusion_kind(), fusion_kind);
|
| parallel_task_assignment.cc | 145 instruction->fusion_kind() != HloInstruction::FusionKind::kLoop) ||
|
| dot_op_emitter.cc | [all...] |
| ir_emitter.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| hlo_parser.cc | 910 optional<HloInstruction::FusionKind> fusion_kind; local
[all...] |
