- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for HasTf2XlaFallback (0.19 sec)
-
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) -
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)