Home
last modified time | relevance | path

Searched refs:dtype_hint (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_arrays.py30 def convert_to_tensor(value, dtype=None, dtype_hint=None): argument
49 elif dtype is None and dtype_hint is None and isinstance(value, float):
51 return ops.convert_to_tensor(value, dtype=dtype, dtype_hint=dtype_hint)
Dnp_array_ops.py135 shape = asarray(np_arrays.convert_to_tensor(shape, dtype_hint=np.int32))
178 result_t = np_arrays.convert_to_tensor(result_t, dtype_hint=dtype)
/external/tensorflow/tensorflow/python/saved_model/
Dfunction_deserialization.py74 ops.convert_to_tensor(arg, dtype_hint=expected.dtype))
81 def _try_convert_to_tensor_spec(arg, dtype_hint): argument
86 result = ops.convert_to_tensor(arg, dtype_hint=dtype_hint)
103 arg = _try_convert_to_tensor_spec(arg, dtype_hint=expected.dtype)
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinalg_impl.py767 a = ops.convert_to_tensor(a, dtype_hint=dtypes.float32, name='a')
961 lower_upper, dtype_hint=dtypes.float32, name='lower_upper')
962 perm = ops.convert_to_tensor(perm, dtype_hint=dtypes.int32, name='perm')
963 rhs = ops.convert_to_tensor(rhs, dtype_hint=lower_upper.dtype, name='rhs')
1063 lower_upper, dtype_hint=dtypes.float32, name='lower_upper')
1064 perm = ops.convert_to_tensor(perm, dtype_hint=dtypes.int32, name='perm')
1115 lower_upper, dtype_hint=dtypes.float32, name='lower_upper')
1116 perm = ops.convert_to_tensor(perm, dtype_hint=dtypes.int32, name='perm')
Dlinear_operator_util.py40 def convert_nonref_to_tensor(value, dtype=None, dtype_hint=None, name=None): argument
118 value, dtype=dtype, dtype_hint=dtype_hint, name=name)
/external/tensorflow/tensorflow/python/ops/
Dstateless_random_ops.py311 probs, dtype_hint=dtypes.float32, name="probs")
313 counts, dtype_hint=probs.dtype, name="counts")
Dmath_ops.py1303 y = ops.convert_to_tensor(y, dtype_hint=x.dtype.base_dtype, name="y")
1523 y = ops.convert_to_tensor(y, dtype_hint=x.dtype.base_dtype, name="y")
3390 b = ops.convert_to_tensor(b, dtype_hint=a.dtype.base_dtype, name="b")
3393 b = ops.convert_to_tensor(b, dtype_hint=a.dtype.base_dtype, name="b")
Darray_ops.py4299 on_value = ops.convert_to_tensor(on_value, dtype_hint=dtype)
4301 off_value = ops.convert_to_tensor(off_value, dtype_hint=dtype)
/external/tensorflow/tensorflow/python/framework/
Dops.py1288 dtype_hint=None): argument
1340 dtype_hint=dtype_hint)
1347 dtype_hint=None): argument
1350 "dtype_hint", dtype_hint, "preferred_dtype", preferred_dtype)
1357 value, dtype=None, dtype_hint=None, name=None): argument
1419 value, dtype=dtype, dtype_hint=dtype_hint, name=name)
1422 def convert_to_tensor_v2(value, dtype=None, dtype_hint=None, name=None): argument
1428 preferred_dtype=dtype_hint,
1501 dtype_hint=None, argument
1506 preferred_dtype = preferred_dtype or dtype_hint
/external/tensorflow/tensorflow/python/data/ops/
Dreaders.py64 filenames = ops.convert_to_tensor(filenames, dtype_hint=dtypes.string)
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor.py2546 value=value, dtype=dtype, dtype_hint=preferred_dtype, name=name)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.pbtxt577 …argspec: "args=[\'value\', \'dtype\', \'dtype_hint\', \'name\'], varargs=None, keywords=None, defa…
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc3396 tensorflow::DataType dtype_hint = dtype_hint_getter(); local
3399 op_exec_info.ctx, input, dtype_hint, op_exec_info.device_name);
Dfunction.py2843 value, dtype_hint=spec.dtype)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt1037 …argspec: "args=[\'value\', \'dtype\', \'name\', \'preferred_dtype\', \'dtype_hint\'], varargs=None…