Searched refs:constant_fold_hook_ (Results 1 – 2 of 2) sorted by relevance
122 constant_fold_hook_ = std::move(fn); in RegisterConstantFoldHook()127 if (constant_fold_hook_) return constant_fold_hook_(op, operands, results); in constantFold()146 static ConstantFoldHook constant_fold_hook_; variable
207 TensorFlowDialect::ConstantFoldHook TensorFlowDialect::constant_fold_hook_; member in mlir::TF::TensorFlowDialect