- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for SelectV2Op (0.12 sec)
-
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
// This specialization is for TF SelectV2 op. SelectV2 op have three inputs and // they should have broadcastable shapes. template <> class ApplyExplicitBroadcasting<TF::SelectV2Op> : public OpRewritePattern<TF::SelectV2Op> { public: using OpRewritePattern<TF::SelectV2Op>::OpRewritePattern; LogicalResult rewriteOpWithDynamicInput(Operation* op, PatternRewriter& rewriter) const {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc
TypeID::get<TF::SegmentSumV2Op>(), TypeID::get<TF::SegmentProdV2Op>(), TypeID::get<TF::SegmentMinV2Op>(), TypeID::get<TF::SegmentMaxV2Op>(), TypeID::get<TF::SelectV2Op>(), TypeID::get<TF::SelfAdjointEigV2Op>(), TypeID::get<TF::SeluGradOp>(), TypeID::get<TF::SeluOp>(), TypeID::get<TF::SigmoidGradOp>(), TypeID::get<TF::SinOp>(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 21.7K bytes - Viewed (0)