Searched defs:final_op (Results 1 – 7 of 7) sorted by relevance
137 StringRef final_op, Operation* op) { in ConvertAllReduce()215 StringRef merge_op, final_op; in matchAndRewrite() local
24 final_op='Id', argument100 final_op='Id', argument
53 final_op='Div', argument
393 const std::string final_op("Id"); in CollectiveSum() local483 const std::string final_op("Id"); in RegisterCollectiveMulFunction() local
93 mlir::Operation** final_op) { in EmitAllReduceForXla()225 mlir::Operation* final_op; in LowerAllReduceOpImpl() local
106 auto final_op = GetOpWithInput(*model, next_op->outputs[0]); in ResolveDilatedConv() local
155 OpKernel* final_op = nullptr; // reduction only member