Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for HasTf2XlaFallback (0.19 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc

              TypeID::get<mlir::TF::WhereOp>(),
              TypeID::get<mlir::TF::XlaSetDynamicDimensionSizeOp>(),
          };
      return *ops;
    }
    
    }  // namespace
    
    bool HasTf2XlaFallback(const TypeID& type_id) {
      return IsOpTypeAllowedTf2XlaFallback(type_id) ||
             IsOpTypeAllowedTf2XlaPreferred(type_id);
    }
    
    bool IsOpLegalizedWithMlir(Operation& op) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 21.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc

      if (HasStringResult(op)) return false;
      if (MatchesPattern(op, supported_ops)) return true;
    
      auto abstractOp = op.getRegisteredInfo();
      if (!abstractOp) return false;
      return mlir::mhlo::HasTf2XlaFallback(abstractOp->getTypeID());
    }
    
    bool IsVariant(Value value) {
      return mlir::isa<mlir::TF::VariantType>(
          getElementTypeOrSelf(value.getType()));
    }
    
    bool HasOutsideCompiledAncestor(Operation* op) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.4K bytes
    - Viewed (0)
Back to top