Home
last modified time | relevance | path

Searched refs:preferred_dtype (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dconvert_to_tensor_or_ragged_tensor_op_test.py41 dict(pylist=[[1, 2], [3]], preferred_dtype=dtypes.float32),
42 dict(pylist=[[1, 2], [3]], preferred_dtype=dtypes.string),
47 preferred_dtype=dtypes.string),
48 dict(pylist=np.array([[1, 2], [3]]), preferred_dtype=dtypes.float32),
49 dict(pylist=np.array([[1, 2], [3]]), preferred_dtype=dtypes.string),
52 preferred_dtype=dtypes.float32),
53 dict(pylist=[np.array(1)], preferred_dtype=dtypes.string),
55 def testConvertRaggedTensor(self, pylist, dtype=None, preferred_dtype=None): argument
58 rt, dtype, preferred_dtype)
82 preferred_dtype=None): argument
[all …]
Dsegment_id_ops.py58 preferred_dtype=dtypes.int64)
Dragged_array_ops.py592 partitions, name='partitions', preferred_dtype=row_splits_dtype)
594 num_partitions, name='num_partitions', preferred_dtype=partitions.dtype)
Dragged_tensor.py969 partition, preferred_dtype=dtypes.int64,
2306 preferred_dtype=None, argument
2339 preferred_dtype=preferred_dtype,
2345 value=value, dtype=dtype, preferred_dtype=preferred_dtype, name=name)
Dragged_dispatch.py200 x = ops.convert_to_tensor(x, name=self._x, preferred_dtype=y.dtype)
202 y = ops.convert_to_tensor(y, name=self._y, preferred_dtype=x.dtype)
Dragged_tensor_shape.py168 ops.convert_to_tensor(size, preferred_dtype=dtypes.int64,
/external/tensorflow/tensorflow/python/ops/distributions/
Ddistribution.py132 def _convert_to_tensor(value, name=None, preferred_dtype=None): argument
135 if (context.executing_eagerly() and preferred_dtype is not None and
136 (preferred_dtype.is_integer or preferred_dtype.is_bool)):
141 value, name=name, preferred_dtype=preferred_dtype)
775 value, name="value", preferred_dtype=self.dtype)
804 value, name="value", preferred_dtype=self.dtype)
833 value, name="value", preferred_dtype=self.dtype)
872 value, name="value", preferred_dtype=self.dtype)
908 value, name="value", preferred_dtype=self.dtype)
948 value, name="value", preferred_dtype=self.dtype)
[all …]
/external/tensorflow/tensorflow/python/ops/
Dtensor_array_ops.py269 value, preferred_dtype=self._dtype, name="value")
336 value, preferred_dtype=self._dtype, name="value")
534 value, preferred_dtype=self._dtype, name="value")
590 value, preferred_dtype=self._dtype, name="value")
604 value, preferred_dtype=self._dtype, name="value")
618 value, preferred_dtype=self._dtype, name="value")
813 value, preferred_dtype=self._dtype, name="value")
903 value, preferred_dtype=self._dtype, name="value")
Darray_ops.py4190 condition, preferred_dtype=dtypes.bool, name="condition")
4287 condition, preferred_dtype=dtypes.bool, name="condition")
5354 tensor = ops.convert_to_tensor(tensor, name=name, preferred_dtype=dtype)
/external/tensorflow/tensorflow/python/framework/
Dops.py1174 preferred_dtype=None, argument
1225 preferred_dtype = deprecation.deprecated_argument_lookup(
1226 "dtype_hint", dtype_hint, "preferred_dtype", preferred_dtype)
1227 return convert_to_tensor_v2(value, dtype, preferred_dtype, name)
1292 preferred_dtype=dtype_hint,
1304 preferred_dtype=None, argument
1310 preferred_dtype = preferred_dtype or dtype_hint
1330 if preferred_dtype is not None:
1331 preferred_dtype = dtypes.as_dtype(preferred_dtype)
1336 if dtype is None and preferred_dtype is not None:
[all …]
Dop_def_library.py413 preferred_dtype=default_dtype,
470 preferred_dtype=default_dtype)
Dops_test.py673 tensor = ops.convert_to_tensor(values, preferred_dtype=dtypes.float32)
678 tensor = ops.convert_to_tensor(values, preferred_dtype=dtypes.int64)
684 tensor = ops.convert_to_tensor(values, preferred_dtype=dtypes.int64)
/external/tensorflow/tensorflow/python/data/util/
Dconvert.py61 ret = ops.convert_to_tensor(shape_like, preferred_dtype=dtypes.int64)
/external/tensorflow/tensorflow/python/ops/signal/
Dfft_ops.py119 preferred_dtype=_dtypes.float32)
149 preferred_dtype=_dtypes.complex64)
/external/tensorflow/tensorflow/python/eager/
Dexecute.py263 t, dtype, preferred_dtype=default_dtype, ctx=ctx))
Dtensor_test.py291 ops.convert_to_tensor(0.5, preferred_dtype=dtypes.int32).dtype,
294 ops.convert_to_tensor(0.5, preferred_dtype=dtypes.float64).dtype,
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py3740 ret = ops.convert_to_tensor(padded_shape, preferred_dtype=dtypes.int64)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt1037 …argspec: "args=[\'value\', \'dtype\', \'name\', \'preferred_dtype\', \'dtype_hint\'], varargs=None…