Searched refs:bounded_compute_graph_mapping (Results 1 – 2 of 2) sorted by relevance
29 bounded_compute_graph_mapping = module.bounded_compute_graph_mapping variable80 shape_compute_graph_mapping.values(), *bounded_compute_graph_mapping.values()143 for schema, (lower_func, upper_func) in bounded_compute_graph_mapping.items():
1194 bounded_compute_graph_mapping: Dict[str, Tuple[ScriptFn, ScriptFn]] = {} variable1223 bounded_compute_graph_mapping[operator_schema] = fns