Searched refs:default_dtype (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | input_spec.py | 281 def to_tensor_spec(input_spec, default_dtype=None): argument
283 default_dtype = default_dtype or backend.floatx()
285 dtype = input_spec.dtype or default_dtype
287 return tensor_spec.TensorSpec(None, default_dtype)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_api_parameter_converter.cc | 160 DataType default_dtype = DT_INVALID) { in ConvertToTensorInPlace() argument
172 if (default_dtype == DT_INVALID) { in ConvertToTensorInPlace()
179 dtype = default_dtype; in ConvertToTensorInPlace()
283 index, &input.ok_dtypes, input.default_dtype)); in ConvertInputsWithTypeAttr()
295 input.default_dtype)); in ConvertInputsWithTypeAttr()
301 dtype = input.default_dtype; in ConvertInputsWithTypeAttr()
376 DataType default_dtype = i < input.default_dtypes.size() in ConvertInputsWithTypeListAttr() local
381 &input.ok_dtypes, default_dtype)); in ConvertInputsWithTypeListAttr()
|
| D | op_def_library.py | 413 default_dtype = None
428 default_dtype = default_type_attr_map[input_arg.type_attr]
437 preferred_dtype=default_dtype,
477 default_dtype = None
487 default_dtype = default_type_attr_map[input_arg.type_attr]
515 preferred_dtype=default_dtype)
522 preferred_dtype=default_dtype)
|
| D | python_api_info.cc | 252 DataType default_dtype = attr_def.has_default_value() in InitializeAttribute() local
255 inputs_with_type_attrs_.push_back({&attr, default_dtype}); in InitializeAttribute()
439 if (input.default_dtype != DT_INVALID) { in DebugInfo()
441 ", default_dtype=", DataType_Name(input.default_dtype)); in DebugInfo()
|
| D | python_api_info.h | 96 DataType default_dtype; // DT_INVALID if no default. member
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | execute.py | 236 def args_to_matching_eager(l, ctx, allowed_dtypes, default_dtype=None): argument
238 if (not l) and (default_dtype is not None):
239 return default_dtype, [] # List is empty; assume default dtype.
274 t, dtype, preferred_dtype=default_dtype, ctx=ctx)
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | stateful_random_ops.cc | 35 #define REGISTER_STATEFUL_OP(name, default_dtype) \ argument
41 .Attr("dtype : type = " #default_dtype) \
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_random_test.py | 194 default_dtype = (
196 self.assertEqual(output.dtype.as_numpy_dtype, default_dtype)
|