Searched defs:tf_dtype (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tensor_array_ops_test.py | 79 def _testTensorArrayWritePack(self, tf_dtype): argument
117 def _testTensorArrayWriteConcat(self, tf_dtype): argument
141 def _testTensorArrayUnpackRead(self, tf_dtype): argument
205 def _testTensorArraySplitRead(self, tf_dtype): argument
1025 def _testTensorArrayScatterRead(self, tf_dtype): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tensor_array_ops_test.py | 52 def _make_converter(tf_dtype): argument
106 def _testTensorArrayWritePack(self, tf_dtype): argument
191 def _testTensorArrayWriteConcat(self, tf_dtype): argument
263 def _testTensorArrayUnpackRead(self, tf_dtype): argument
313 def _testTensorArraySplitRead(self, tf_dtype): argument
|
| D | confusion_matrix_test.py | 81 def _testConfMatrixOnTensors(self, tf_dtype, np_dtype): argument
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | dlpack.cc | 150 TF_DataType* tf_dtype) { in TfDataTypeFormDlDataType()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | util.py | 96 def convert_dtype_to_tflite_type(tf_dtype): argument
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 1090 DataType tf_dtype; in GetTempWeights() local
6408 DataType tf_dtype = attrs.get<DataType>("T"); in ConvertGraphDefToEngine() local
|