| /external/tensorflow/tensorflow/c/eager/ |
| D | dlpack.cc | 150 TF_DataType* tf_dtype) { in TfDataTypeFormDlDataType() argument
155 *tf_dtype = TF_DataType::TF_UINT8; in TfDataTypeFormDlDataType()
158 *tf_dtype = TF_DataType::TF_UINT16; in TfDataTypeFormDlDataType()
161 *tf_dtype = TF_DataType::TF_UINT32; in TfDataTypeFormDlDataType()
164 *tf_dtype = TF_DataType::TF_UINT64; in TfDataTypeFormDlDataType()
174 *tf_dtype = TF_DataType::TF_INT8; in TfDataTypeFormDlDataType()
177 *tf_dtype = TF_DataType::TF_INT16; in TfDataTypeFormDlDataType()
180 *tf_dtype = TF_DataType::TF_INT32; in TfDataTypeFormDlDataType()
183 *tf_dtype = TF_DataType::TF_INT64; in TfDataTypeFormDlDataType()
193 *tf_dtype = TF_DataType::TF_HALF; in TfDataTypeFormDlDataType()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tensor_array_ops_test.py | 79 def _testTensorArrayWritePack(self, tf_dtype): argument
81 convert = _make_converter(tf_dtype)
85 dtype=tf_dtype, tensor_array_name="foo", size=3)
117 def _testTensorArrayWriteConcat(self, tf_dtype): argument
119 convert = _make_converter(tf_dtype)
123 dtype=tf_dtype, tensor_array_name="foo", size=3)
141 def _testTensorArrayUnpackRead(self, tf_dtype): argument
143 convert = _make_converter(tf_dtype)
147 dtype=tf_dtype, tensor_array_name="foo", size=3)
164 dtype=tf_dtype, tensor_array_name="foo", size=3)
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/signal/ |
| D | window_ops_test.py | 90 tf_dtype, tol = tf_dtype_tol
91 np_dtype = tf_dtype.as_numpy_dtype
95 dtype=tf_dtype)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | confusion_matrix_test.py | 81 def _testConfMatrixOnTensors(self, tf_dtype, np_dtype): argument
93 data = math_ops.cast(math_ops.round(data), tf_dtype)
98 [20], dtype=tf_dtype), array_ops.ones(
99 [20], dtype=tf_dtype)
104 lab, data, dtype=tf_dtype, num_classes=2)
|
| D | tensor_array_ops_test.py | 52 def _make_converter(tf_dtype): argument
54 if tf_dtype == dtypes.string:
57 x = np.asarray(x).astype(tf_dtype.as_numpy_dtype)
58 if tf_dtype.is_complex:
106 def _testTensorArrayWritePack(self, tf_dtype): argument
109 dtype=tf_dtype, tensor_array_name="foo", size=3)
111 convert = _make_converter(tf_dtype)
191 def _testTensorArrayWriteConcat(self, tf_dtype): argument
194 dtype=tf_dtype, tensor_array_name="foo", size=3, infer_shape=False)
196 convert = _make_converter(tf_dtype)
[all …]
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | util.py | 96 def convert_dtype_to_tflite_type(tf_dtype): argument
108 result = _MAP_TF_TO_TFLITE_TYPES.get(tf_dtype)
110 raise ValueError("Unsupported tf.dtype {0}".format(tf_dtype))
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 1598 tf_dtype = dtypes_module.as_dtype(dtype)
1599 v = array_ops.zeros(shape=shape, dtype=tf_dtype, name=name)
1634 tf_dtype = dtypes_module.as_dtype(dtype)
1635 v = array_ops.ones(shape=shape, dtype=tf_dtype, name=name)
1668 tf_dtype = dtypes_module.as_dtype(dtype)
1669 return variable(linalg_ops.eye(size, dtype=tf_dtype), dtype, name)
1765 tf_dtype = dtypes_module.as_dtype(dtype)
1770 low, high, dtype=tf_dtype, seed=seed)(shape)
1801 tf_dtype = dtypes_module.as_dtype(dtype)
1806 mean, scale, dtype=tf_dtype, seed=seed)(shape)
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 1090 DataType tf_dtype; in GetTempWeights() local
1093 TF_CHECK_OK(TrtTypeToTfType(trt_dtype, &tf_dtype)); in GetTempWeights()
1095 Tensor tensor(tf_dtype, shape); in GetTempWeights()
6408 DataType tf_dtype = attrs.get<DataType>("T"); in ConvertGraphDefToEngine() local
6410 TF_RETURN_IF_ERROR(TfTypeToTrtType(tf_dtype, &trt_dtype)); in ConvertGraphDefToEngine()
|