Searched refs:shape_compute_graph_mapping (Results 1 – 2 of 2) sorted by relevance
30 shape_compute_graph_mapping = module.shape_compute_graph_mapping variable80 shape_compute_graph_mapping.values(), *bounded_compute_graph_mapping.values()130 for schema, scripted_func in shape_compute_graph_mapping.items():
1193 shape_compute_graph_mapping: Dict[str, ScriptFn] = {} variable1213 global shape_compute_graph_mapping1215 shape_compute_graph_mapping[operator_schema] = process_func(func)