Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/transforms/consolidate_attrs/
Dpass.cc217 class ReifyOperationOutputShapes : public RewritePattern { class
219 ReifyOperationOutputShapes(MLIRContext *context, PatternBenefit benefit, in ReifyOperationOutputShapes() function in mlir::tfg::__anonc9ca0e7d0311::ReifyOperationOutputShapes
265 class ReifyTFGOpOutputShapes : public ReifyOperationOutputShapes {
268 : ReifyOperationOutputShapes(context, /*benefit=*/1, "_output_shapes"), in ReifyTFGOpOutputShapes()
290 struct ReifyCFOpOutputShapes : public ReifyOperationOutputShapes {
294 : ReifyOperationOutputShapes(context, /*benefit=*/2, "output_shapes") {} in ReifyCFOpOutputShapes()