Searched defs:default_dtype (Results 1 – 6 of 6) sorted by relevance
35 #define REGISTER_STATEFUL_OP(name, default_dtype) \ argument
277 def to_tensor_spec(input_spec, default_dtype=None): argument
160 DataType default_dtype = DT_INVALID) { in ConvertToTensorInPlace()376 DataType default_dtype = i < input.default_dtypes.size() in ConvertInputsWithTypeListAttr() local
96 DataType default_dtype; // DT_INVALID if no default. member
252 DataType default_dtype = attr_def.has_default_value() in InitializeAttribute() local
225 def args_to_matching_eager(l, ctx, allowed_dtypes, default_dtype=None): argument