| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tensor_array_ops_test.py | 73 def _testTensorArrayWritePack(self, tf_dtype): argument
76 dtype=tf_dtype, tensor_array_name="foo", size=3)
78 convert = _make_converter(tf_dtype)
108 def _testTensorArrayWriteConcat(self, tf_dtype): argument
111 dtype=tf_dtype, tensor_array_name="foo", size=3)
113 convert = _make_converter(tf_dtype)
130 def _testTensorArrayUnpackRead(self, tf_dtype): argument
133 dtype=tf_dtype, tensor_array_name="foo", size=3)
135 convert = _make_converter(tf_dtype)
149 dtype=tf_dtype, tensor_array_name="foo", size=3)
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/signal/ |
| D | window_ops_test.py | 71 for tf_dtype, tol in self._dtypes:
72 np_dtype = tf_dtype.as_numpy_dtype
76 dtype=tf_dtype).eval()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert.py | 82 def convert_dtype_to_tflite_type(tf_dtype): argument
94 result = _MAP_TF_TO_TFLITE_TYPES.get(tf_dtype)
96 raise ValueError("Unsupported tf.dtype {0}".format(tf_dtype))
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tensor_array_ops_test.py | 49 def _make_converter(tf_dtype): argument
51 if tf_dtype == dtypes.string:
54 x = np.asarray(x).astype(tf_dtype.as_numpy_dtype)
55 if tf_dtype.is_complex:
103 def _testTensorArrayWritePack(self, tf_dtype): argument
106 dtype=tf_dtype, tensor_array_name="foo", size=3)
108 convert = _make_converter(tf_dtype)
147 def _testTensorArrayWriteConcat(self, tf_dtype): argument
150 dtype=tf_dtype, tensor_array_name="foo", size=3, infer_shape=False)
152 convert = _make_converter(tf_dtype)
[all …]
|
| 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)
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 1081 tf_dtype = dtypes_module.as_dtype(dtype)
1082 v = array_ops.zeros(shape=shape, dtype=tf_dtype, name=name)
1116 tf_dtype = dtypes_module.as_dtype(dtype)
1117 v = array_ops.ones(shape=shape, dtype=tf_dtype, name=name)
1149 tf_dtype = dtypes_module.as_dtype(dtype)
1150 return variable(linalg_ops.eye(size, dtype=tf_dtype), dtype, name)
1246 tf_dtype = dtypes_module.as_dtype(dtype)
1251 low, high, dtype=tf_dtype, seed=seed)(shape)
1284 tf_dtype = dtypes_module.as_dtype(dtype)
1289 mean, scale, dtype=tf_dtype, seed=seed)(shape)
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 86 nvinfer1::DataType TfDataTypeToTrt(DataType tf_dtype) { in TfDataTypeToTrt() argument
87 switch (tf_dtype) { in TfDataTypeToTrt()
95 QCHECK(false) << "Unexpected data type " << DataTypeString(tf_dtype); in TfDataTypeToTrt()
1108 const DataType tf_dtype = TrtDataTypeToTf(trt_dtype); in CheckDataTypeMatches() local
1109 ASSERT_EQ(data.tensor.dtype(), tf_dtype) in CheckDataTypeMatches()
1111 << DataTypeString(tf_dtype); in CheckDataTypeMatches()
1197 DataType tf_dtype = TrtDataTypeToTf(trt_dtype); in AddTestTensor() local
1201 auto input = ops::Placeholder(scope_.WithOpName(name), tf_dtype, attrs); in AddTestTensor()
|
| D | convert_nodes.cc | 97 inline Status ConvertDType(DataType tf_dtype, nvinfer1::DataType* trt_dtype) { in ConvertDType() argument
98 switch (tf_dtype) { in ConvertDType()
114 DataTypeString(tf_dtype)); in ConvertDType()
4138 DataType tf_dtype = attrs.get<DataType>("T"); in ConvertGraphDefToEngine() local
4140 TF_RETURN_IF_ERROR(ConvertDType(tf_dtype, &trt_dtype)); in ConvertGraphDefToEngine()
|