Searched refs:parent_func (Results 1 – 5 of 5) sorted by relevance
106 const FunctionDef* parent_func = in OptimizeAndCollectStats() local111 if (!fusion_utils::CanCompose(parent_func->signature(), in OptimizeAndCollectStats()119 *parent_func, *func, "fused_map", fusion_utils::ComposeSignature, in OptimizeAndCollectStats()
83 auto parent_func = island_op->getParentOfType<func::FuncOp>(); in runOnOperation() local85 parent_func->getAttrOfType<BoolAttr>(mlir::TF::kSkipIslandOutlining); in runOnOperation()
578 parent_func = self.file_module_function_of(frame.f_back)579 self._callers[(parent_func, this_func)] = 1
516 parent_func = self.file_module_function_of(frame.f_back)517 self._callers[(parent_func, this_func)] = 1
2182 mlir::func::FuncOp parent_func = in runOnOperation() local2184 int64_t max_arg_size = get_max_arg_size(parent_func); in runOnOperation()