Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/tpu/
Dbfloat16.py54 requested_dtype = kwargs['dtype']
55 if requested_dtype == dtypes.bfloat16:
/external/tensorflow/tensorflow/python/lib/core/
Dpy_seq_tensor.cc771 DataType requested_dtype = DT_INVALID; in PySeqToTFE_TensorHandle() local
773 requested_dtype = dtype; in PySeqToTFE_TensorHandle()
784 switch (requested_dtype) { in PySeqToTFE_TensorHandle()
829 if (requested_dtype == DT_INVALID) { in PySeqToTFE_TensorHandle()
851 if (requested_dtype == DT_INVALID) { in PySeqToTFE_TensorHandle()
884 requested_dtype == DT_INVALID ? DT_FLOAT : requested_dtype, in PySeqToTFE_TensorHandle()