Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_arrays.py24 def convert_to_tensor(value, dtype=None, dtype_hint=None): argument
43 elif dtype is None and dtype_hint is None and isinstance(value, float):
45 return ops.convert_to_tensor(value, dtype=dtype, dtype_hint=dtype_hint)
Dnp_array_ops.py122 shape = asarray(np_arrays.convert_to_tensor(shape, dtype_hint=np.int32))
164 result_t = np_arrays.convert_to_tensor(result_t, dtype_hint=dtype)
/external/tensorflow/tensorflow/python/ops/ragged/
Drow_partition.py197 dtype_hint=None): argument
249 value_rowids, "value_rowids", dtype_hint=dtype_hint, dtype=dtype)
316 dtype_hint=None): argument
359 row_splits, "row_splits", dtype_hint=dtype_hint, dtype=dtype)
379 dtype_hint=None): argument
411 row_lengths, "row_lengths", dtype_hint=dtype_hint, dtype=dtype)
435 dtype_hint=None): argument
462 row_starts, "row_starts", dtype_hint=dtype_hint, dtype=dtype)
486 dtype_hint=None): argument
511 row_limits, "row_limits", dtype_hint=dtype_hint, dtype=dtype)
[all …]
Dragged_tensor.py408 dtype_hint=_get_optional_partition_dtype(values))
456 dtype_hint=_get_optional_partition_dtype(values))
500 dtype_hint=_get_optional_partition_dtype(values))
541 dtype_hint=_get_optional_partition_dtype(values))
580 dtype_hint=_get_optional_partition_dtype(values))
656 dtype_hint=_get_optional_partition_dtype(values))
2691 dtype_hint=preferred_dtype,
2697 value=value, dtype=dtype, dtype_hint=preferred_dtype, name=name)
Ddynamic_ragged_shape.py247 inner_shape, dtype_hint=dtype, name="inner_dim_sizes")
3103 current_lengths, nvals, nrows=nrows, dtype_hint=dtype))
3108 RowPartition.from_row_lengths(current_lengths, dtype_hint=dtype))
Ddynamic_ragged_shape_test.py1554 1, nvals=1, nrows=1, dtype_hint=dtypes.int64)
1555 b_1 = RowPartition.from_row_lengths([2, 1, 3], dtype_hint=dtypes.int64)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.experimental.-row-partition.pbtxt28 …argspec: "args=[\'cls\', \'row_lengths\', \'validate\', \'dtype\', \'dtype_hint\'], varargs=None, …
32 …argspec: "args=[\'cls\', \'row_limits\', \'validate\', \'dtype\', \'dtype_hint\'], varargs=None, k…
36 …argspec: "args=[\'cls\', \'row_splits\', \'validate\', \'dtype\', \'dtype_hint\'], varargs=None, k…
40 …argspec: "args=[\'cls\', \'row_starts\', \'nvals\', \'validate\', \'dtype\', \'dtype_hint\'], vara…
44 …uniform_row_length\', \'nvals\', \'nrows\', \'validate\', \'dtype\', \'dtype_hint\'], varargs=None…
48 …argspec: "args=[\'cls\', \'value_rowids\', \'nrows\', \'validate\', \'dtype\', \'dtype_hint\'], va…
Dtensorflow.pbtxt1045 …argspec: "args=[\'value\', \'dtype\', \'name\', \'preferred_dtype\', \'dtype_hint\'], varargs=None…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.experimental.-row-partition.pbtxt28 …argspec: "args=[\'cls\', \'row_lengths\', \'validate\', \'dtype\', \'dtype_hint\'], varargs=None, …
32 …argspec: "args=[\'cls\', \'row_limits\', \'validate\', \'dtype\', \'dtype_hint\'], varargs=None, k…
36 …argspec: "args=[\'cls\', \'row_splits\', \'validate\', \'dtype\', \'dtype_hint\'], varargs=None, k…
40 …argspec: "args=[\'cls\', \'row_starts\', \'nvals\', \'validate\', \'dtype\', \'dtype_hint\'], vara…
44 …uniform_row_length\', \'nvals\', \'nrows\', \'validate\', \'dtype\', \'dtype_hint\'], varargs=None…
48 …argspec: "args=[\'cls\', \'value_rowids\', \'nrows\', \'validate\', \'dtype\', \'dtype_hint\'], va…
Dtensorflow.pbtxt581 …argspec: "args=[\'value\', \'dtype\', \'dtype_hint\', \'name\'], varargs=None, keywords=None, defa…
/external/pytorch/torch/_dynamo/
Ddebug_utils.py501 def storage(self, storage_hash, nbytes, *, device=None, dtype_hint=None): argument
525 def storage(self, storage_hash, nbytes, *, device=None, dtype_hint=None): argument
529 dtype_hint = _dtype_or_default(dtype_hint)
543 shape = (nbytes // dtype_hint.itemsize,)
545 return rand_strided(shape, stride, dtype_hint, device).untyped_storage()
625 def storage(self, untyped_storage, *, dtype_hint=None, device_hint=None) -> str: argument
632 if _dtype_or_default(None) != _dtype_or_default(dtype_hint):
657 t.untyped_storage(), dtype_hint=t.dtype, device_hint=t.device
/external/tensorflow/tensorflow/python/saved_model/
Dfunction_deserialization.py73 ops.convert_to_tensor(arg, dtype_hint=expected.dtype))
82 def _try_convert_to_tensor_spec(arg, dtype_hint): argument
87 result = ops.convert_to_tensor(arg, dtype_hint=dtype_hint)
104 arg = _try_convert_to_tensor_spec(arg, dtype_hint=expected.dtype)
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_util.py36 def convert_nonref_to_tensor(value, dtype=None, dtype_hint=None, name=None): argument
115 value, dtype=dtype, dtype_hint=dtype_hint, name=name)
Dlinalg_impl.py786 a = ops.convert_to_tensor(a, dtype_hint=dtypes.float32, name='a')
980 lower_upper, dtype_hint=dtypes.float32, name='lower_upper')
981 perm = ops.convert_to_tensor(perm, dtype_hint=dtypes.int32, name='perm')
982 rhs = ops.convert_to_tensor(rhs, dtype_hint=lower_upper.dtype, name='rhs')
1082 lower_upper, dtype_hint=dtypes.float32, name='lower_upper')
1083 perm = ops.convert_to_tensor(perm, dtype_hint=dtypes.int32, name='perm')
1134 lower_upper, dtype_hint=dtypes.float32, name='lower_upper')
1135 perm = ops.convert_to_tensor(perm, dtype_hint=dtypes.int32, name='perm')
/external/tensorflow/tensorflow/python/framework/
Dops.py1360 dtype_hint=None): argument
1412 dtype_hint=dtype_hint)
1419 dtype_hint=None): argument
1422 "dtype_hint", dtype_hint, "preferred_dtype", preferred_dtype)
1429 value, dtype=None, dtype_hint=None, name=None): argument
1491 value, dtype=dtype, dtype_hint=dtype_hint, name=name)
1494 def convert_to_tensor_v2(value, dtype=None, dtype_hint=None, name=None): argument
1500 preferred_dtype=dtype_hint,
1574 dtype_hint=None, argument
1579 preferred_dtype = preferred_dtype or dtype_hint
/external/tensorflow/tensorflow/python/ops/
Dstateless_random_ops.py554 probs, dtype_hint=dtypes.float32, name="probs")
556 counts, dtype_hint=probs.dtype, name="counts")
Dmath_ops.py1513 y = ops.convert_to_tensor(y, dtype_hint=x.dtype.base_dtype, name="y")
1753 y = ops.convert_to_tensor(y, dtype_hint=x.dtype.base_dtype, name="y")
3633 b = ops.convert_to_tensor(b, dtype_hint=a.dtype.base_dtype, name="b")
3636 b = ops.convert_to_tensor(b, dtype_hint=a.dtype.base_dtype, name="b")
4000 y = ops.convert_to_tensor(y, dtype_hint=x.dtype.base_dtype, name="y")
Darray_ops.py4425 on_value = ops.convert_to_tensor(on_value, dtype_hint=dtype)
4427 off_value = ops.convert_to_tensor(off_value, dtype_hint=dtype)
/external/tensorflow/tensorflow/python/data/ops/
Dreaders.py66 filenames = ops.convert_to_tensor(filenames, dtype_hint=dtypes.string)
/external/tensorflow/tensorflow/python/eager/
Dfunction_spec.py627 value, dtype_hint=spec.dtype)
Dpywrap_tfe_src.cc3479 tensorflow::DataType dtype_hint = dtype_hint_getter(); local
3482 op_exec_info.ctx, input, dtype_hint, op_exec_info.device_name);