Searched defs:tf_dtype (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tensor_array_ops_test.py | 73 def _testTensorArrayWritePack(self, tf_dtype): argument
108 def _testTensorArrayWriteConcat(self, tf_dtype): argument
130 def _testTensorArrayUnpackRead(self, tf_dtype): argument
185 def _testTensorArraySplitRead(self, tf_dtype): argument
943 def _testTensorArrayScatterRead(self, tf_dtype): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tensor_array_ops_test.py | 49 def _make_converter(tf_dtype): argument
103 def _testTensorArrayWritePack(self, tf_dtype): argument
147 def _testTensorArrayWriteConcat(self, tf_dtype): argument
219 def _testTensorArrayUnpackRead(self, tf_dtype): argument
269 def _testTensorArraySplitRead(self, tf_dtype): argument
|
| D | confusion_matrix_test.py | 81 def _testConfMatrixOnTensors(self, tf_dtype, np_dtype): argument
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert.py | 82 def convert_dtype_to_tflite_type(tf_dtype): argument
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 86 nvinfer1::DataType TfDataTypeToTrt(DataType tf_dtype) { in TfDataTypeToTrt()
1108 const DataType tf_dtype = TrtDataTypeToTf(trt_dtype); in CheckDataTypeMatches() local
1197 DataType tf_dtype = TrtDataTypeToTf(trt_dtype); in AddTestTensor() local
|
| D | convert_nodes.cc | 97 inline Status ConvertDType(DataType tf_dtype, nvinfer1::DataType* trt_dtype) { in ConvertDType()
4138 DataType tf_dtype = attrs.get<DataType>("T"); in ConvertGraphDefToEngine() local
|