Searched defs:valid_dtypes (Results 1 – 1 of 1) sorted by relevance
534 def check_tensors_dtypes_same_and_valid(args, valid_dtypes, prim_name): argument541 def check_tensor_dtype_valid(arg_name, arg_type, valid_dtypes, prim_name): argument