Searched refs:dtype_hierarchy (Results 1 – 2 of 2) sorted by relevance
115 def _infer_matching_dtype(tensors, dtype_hierarchy): argument117 assert all(t.dtype in dtype_hierarchy for t in tensors)118 inferred_dtype = max([t.dtype for t in tensors], key=dtype_hierarchy.index)
1876 dtype_hierarchy = [1879 assert all(arg.dtype in dtype_hierarchy for arg in [start, limit, delta])1881 key=dtype_hierarchy.index)