Searched refs:target_dtype (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 398 target_dtype, argument
435 if (not _is_integer_like_by_dtype(target_dtype) and
436 not target_dtype.is_floating):
438 "integer-type.".format(target_dtype.name))
440 not _is_integer_like_by_dtype(target_dtype)):
443 target_dtype.name))
458 int_dtype=target_dtype,
460 target_dtype.name)),
464 _largest_integer_by_dtype(target_dtype)):
469 _largest_integer_by_dtype(target_dtype),
[all …]
|
| D | categorical.py | 288 k, target_dtype=dtypes.int32)
310 k, target_dtype=dtypes.int32)
|
| D | bernoulli.py | 135 event, target_dtype=dtypes.bool)
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | util_test.py | 321 x, target_dtype=dtypes.int16)
330 x, target_dtype=dtypes.int16)
339 x, target_dtype=dtypes.int16)
348 x, target_dtype=dtypes.uint16, assert_nonnegative=False)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_v1.py | 2990 target_dtype = losses.LABEL_DTYPES_FOR_LOSSES.get(
2997 dtype=target_dtype)
|