Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
Dinsert_custom_aggregation_ops.cc38 class InsertCustomAggregationOpsPass class
39 : public PassWrapper<InsertCustomAggregationOpsPass,
42 MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(InsertCustomAggregationOpsPass) in MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID() argument
62 static PassRegistration<InsertCustomAggregationOpsPass> pass;
120 void InsertCustomAggregationOpsPass::runOnOperation() { in runOnOperation()
136 return std::make_unique<InsertCustomAggregationOpsPass>(); in CreateInsertCustomAggregationOpsPass()