Home
last modified time | relevance | path

Searched refs:requested_dtype (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dbfloat16.py50 requested_dtype = kwargs['dtype']
51 if requested_dtype == dtypes.bfloat16:
/external/tensorflow/tensorflow/python/lib/core/
Dpy_seq_tensor.cc497 DataType requested_dtype = DT_INVALID; in PySeqToTensor() local
501 requested_dtype = static_cast<DataType>(dtype_as_int); in PySeqToTensor()
508 switch (requested_dtype) { in PySeqToTensor()
546 if (requested_dtype == DT_INVALID) { in PySeqToTensor()
565 if (requested_dtype == DT_INVALID) { in PySeqToTensor()
594 *ret = Tensor(requested_dtype == DT_INVALID ? DT_FLOAT : requested_dtype, in PySeqToTensor()