Searched defs:allowed_dtypes (Results 1 – 5 of 5) sorted by relevance
115 const auto& allowed_dtypes = Impl::AllowedDataTypes(); in ValidateNodeDefDataType() local
225 def args_to_matching_eager(l, ctx, allowed_dtypes, default_dtype=None): argument
92 dtype_vector allowed_dtypes, in check_tensor_dtype()
1093 allowed_dtypes = [ variable in ReciprocalNoNanTest
78 def __init__(self, module_info_iterable, allowed_dtypes=None, argument