Searched refs:decompose_funcs (Results 1 – 1 of 1) sorted by relevance
87 static bool DecomposeOps(Block* block, CompilationUnit& decompose_funcs) { in DecomposeOps() argument92 DecomposeOps(sub, decompose_funcs); in DecomposeOps()110 toGraphFunction(decompose_funcs.get_function("addmm")).graph(); in DecomposeOps()140 toGraphFunction(decompose_funcs.get_function("batch_norm")).graph(); in DecomposeOps()175 toGraphFunction(decompose_funcs.get_function("layer_norm")).graph(); in DecomposeOps()195 static CompilationUnit decompose_funcs(R"SCRIPT( in DecomposeOps() local224 bool is_decomposed = DecomposeOps(graph->block(), decompose_funcs); in DecomposeOps()