HomeSort by relevance Sort by last modified time
    Searched defs:fused_root (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
ops.h 43 HloInstruction* fused_root = fusion->fused_expression_root(); local
44 if (fused_root->opcode() != HloOpcode::kDynamicUpdateSlice ||
53 fused_root->mutable_operand(0)->LatestNonGteAncestorAndIndex();
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
instruction_fusion_test.cc 159 HloInstruction* fused_root = root->fused_expression_root(); local
160 EXPECT_EQ(HloOpcode::kAdd, fused_root->opcode());
161 // Check that operands of 'fused_root' are GTE.
162 EXPECT_EQ(HloOpcode::kGetTupleElement, fused_root->operand(0)->opcode());
163 EXPECT_EQ(HloOpcode::kGetTupleElement, fused_root->operand(1)->opcode());
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_verifier.cc 632 const HloInstruction* fused_root = fusion->fused_expression_root(); local
635 if (fused_root == instruction) {
665 if (fused_root->user_count() != 0) {
674 if (instruction != fused_root) {
    [all...]
hlo_instruction.cc 779 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root) {
783 instruction->set_parent(fused_root->parent());
784 instruction->set_metadata(fused_root->metadata());
785 instruction->CloneAndFuseInternal(fused_root);
1021 HloInstruction* fused_root = fused_expression_root(); local
    [all...]

Completed in 4279 milliseconds