- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for HasTf2XlaFallback (0.69 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config_test.cc
EXPECT_TRUE(IsTypeLegalizedWithMlir(TypeID::get<TF::ModOp>())); EXPECT_FALSE(HasTf2XlaFallback(TypeID::get<TF::ModOp>())); EXPECT_FALSE(IsOpAllowedTf2xlaFallback(TypeID::get<TF::ModOp>())); EXPECT_FALSE(IsOpAllowedTf2xlaPreferred(TypeID::get<TF::ModOp>())); } TEST_F(LegalizationOpConfigTest, ExpectsTrueForTF2XLATypeID) { EXPECT_TRUE(HasTf2XlaFallback(TypeID::get<TF::AllOp>()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 30 03:31:01 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.h
bool IsDynamicPadderOp(const TypeID& type_id); // Returns True if this op has a Tf2XLA fallback. Currently, this is not the // inverse of the !IsOpLegalizedWithMlir, but it should be. bool HasTf2XlaFallback(const TypeID& type_id); // Whether this type is allowed to have a TF2XLA fallback. bool IsOpAllowedTf2xlaFallback(const TypeID& type_id); // Whether this type is Preferred to use a TF2XLA fallback kernel when using
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 20:53:25 UTC 2024 - 2K bytes - Viewed (0) -
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)