Searched refs:dtype_hierarchy (Results 1 – 2 of 2) sorted by relevance
107 def _infer_matching_dtype(tensors, dtype_hierarchy): argument109 assert all(t.dtype in dtype_hierarchy for t in tensors)110 inferred_dtype = max([t.dtype for t in tensors], key=dtype_hierarchy.index)
1265 dtype_hierarchy = [1268 assert all(arg.dtype in dtype_hierarchy for arg in [start, limit, delta])1271 key=dtype_hierarchy.index)