Searched refs:ReplaceReductionWithIdentity (Results 1 – 3 of 3) sorted by relevance
237 bool ReplaceReductionWithIdentity(NodeDef* node) const;
2861 bool ConstantFolding::ReplaceReductionWithIdentity(NodeDef* node) const { in ReplaceReductionWithIdentity() function in tensorflow::grappler::ConstantFolding2887 return ReplaceReductionWithIdentity(node); in SimplifyReduction()2950 return ReplaceReductionWithIdentity(node); in SimplifyReduction()
1978 Operation *identity_op = ReplaceReductionWithIdentity(rewriter, op); in matchAndRewrite()2034 Operation *identity_op = ReplaceReductionWithIdentity(rewriter, op); in matchAndRewrite()2082 Operation *ReplaceReductionWithIdentity(OpBuilder &builder, in ReplaceReductionWithIdentity() function in mlir::tfg::SimplifyReductionOp