Home
last modified time | relevance | path

Searched refs:target_dtype (Results 1 – 8 of 8) sorted by relevance

/external/armnn/python/pyarmnn/src/pyarmnn/_quantization/
Dquantize_and_dequantize.py24 def quantize(value: float, scale: float, offset: int, target_dtype: str) -> int:
40 if target_dtype not in __DTYPE_TO_QUANTIZE_FUNCTION:
42 …Armnn currently supports quantization to {} values.""".format(target_dtype, list(__DTYPE_TO_QUANTI…
44 return __DTYPE_TO_QUANTIZE_FUNCTION[target_dtype](float(value), scale, offset)
/external/tensorflow/tensorflow/python/ops/distributions/
Dutil.py394 target_dtype, argument
431 if (not _is_integer_like_by_dtype(target_dtype) and
432 not target_dtype.is_floating):
434 "integer-type.".format(target_dtype.name))
436 not _is_integer_like_by_dtype(target_dtype)):
439 target_dtype.name))
454 int_dtype=target_dtype,
456 target_dtype.name)),
460 _largest_integer_by_dtype(target_dtype)):
465 _largest_integer_by_dtype(target_dtype),
[all …]
Dcategorical.py284 k, target_dtype=dtypes.int32)
306 k, target_dtype=dtypes.int32)
Dbernoulli.py131 event, target_dtype=dtypes.bool)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Drejection_resample_test.py154 def testOtherDtypes(self, target_dtype, init_dtype): argument
155 target_dist = np.array([0.5, 0.5], dtype=target_dtype)
/external/tensorflow/tensorflow/python/ops/ragged/
Drow_partition.py1468 target_dtype = _get_target_dtype([x for (x, _) in values], dtype, dtype_hint)
1477 x, dtype=target_dtype, name=name) for (x, name) in values
1481 None if x is None else math_ops.cast(x, dtype=target_dtype, name=name)
/external/tensorflow/tensorflow/python/kernel_tests/distributions/
Dutil_test.py317 x, target_dtype=dtypes.int16)
326 x, target_dtype=dtypes.int16)
335 x, target_dtype=dtypes.int16)
344 x, target_dtype=dtypes.uint16, assert_nonnegative=False)
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_v1.py2998 target_dtype = losses.LABEL_DTYPES_FOR_LOSSES.get(
3005 dtype=target_dtype)