Home
last modified time | relevance | path

Searched refs:ConvertTFEinsumOp (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Deinsum.h43 struct ConvertTFEinsumOp : public OpRewritePattern<TF::EinsumOp> { struct
45 explicit ConvertTFEinsumOp(MLIRContext* context) in ConvertTFEinsumOp() function
Deinsum.cc350 LogicalResult ConvertTFEinsumOp::matchAndRewrite( in matchAndRewrite()
375 patterns.insert<ConvertTFEinsumOp>(&getContext()); in runOnFunction()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_tf.cc1483 phase_2_patterns.insert<TF::ConvertTFEinsumOp, ConvertTFBroadcastTo, in runOnFunction()