Home
last modified time | relevance | path

Searched refs:SigmoidOp (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc1878 auto tf_sigmoid_op = cast<TF::SigmoidOp>(op); in matchAndRewrite()
1883 rewriter.replaceOpWithNewOp<tosa::SigmoidOp>(op, output_type, in matchAndRewrite()
Dlegalize_tfl.cc2619 rewriter.replaceOpWithNewOp<tosa::SigmoidOp>(op, output_type, in matchAndRewrite()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision_test.cc1138 TEST_F(AutoMixedPrecisionTest, SigmoidOp) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2645 class ConvertSigmoidOp : public OpRewritePattern<TF::SigmoidOp> {
2649 LogicalResult matchAndRewrite(TF::SigmoidOp op, in matchAndRewrite()