Searched defs:tf_dtype (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tensor_array_ops_test.py | 75 def _testTensorArrayWritePack(self, tf_dtype): argument
113 def _testTensorArrayWriteConcat(self, tf_dtype): argument
137 def _testTensorArrayUnpackRead(self, tf_dtype): argument
201 def _testTensorArraySplitRead(self, tf_dtype): argument
1021 def _testTensorArrayScatterRead(self, tf_dtype): argument
|
| /external/tensorflow/tensorflow/core/runtime_fallback/util/ |
| D | type_util_test.cc | 28 DataType tf_dtype; member
|
| D | attr_util.cc | 222 DataType tf_dtype = ConvertToTfDataType(op_attr_type); in FillAttrValueMapUsingScalar() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
| D | tensor_array_ops_test.py | 48 def _make_converter(tf_dtype): argument
102 def _testTensorArrayWritePack(self, tf_dtype): argument
187 def _testTensorArrayWriteConcat(self, tf_dtype): argument
259 def _testTensorArrayUnpackRead(self, tf_dtype): argument
309 def _testTensorArraySplitRead(self, tf_dtype): argument
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | weights.cc | 191 DataType tf_dtype; in GetTempWeights() local
|
| D | convert_nodes.cc | 5913 DataType tf_dtype = node_def.attr().at(type_key).type(); in ConvertGraphDefToEngine() local
5969 DataType tf_dtype; in ConvertGraphDefToEngine() local
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | tensor_util.cc | 67 auto tf_dtype = tensorflow::tfd::GetTfDataType(tensor.dtype()); in CopyScalarHostTensorToTFTensor() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| D | confusion_matrix_test.py | 76 def _testConfMatrixOnTensors(self, tf_dtype, np_dtype): argument
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | dlpack.cc | 156 TF_DataType* tf_dtype) { in TfDataTypeFormDlDataType()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_kernels.cc | 447 DataType tf_dtype = ConvertToTfDataType(op_attr_type); in PrepareAttributes() local
1229 auto tf_dtype = tensor_interface->Type(); in ConvertTFTensorHandleToTFRTTensor() local
|
| /external/tensorflow/tensorflow/core/ir/importexport/ |
| D | graphdef_import.cc | 895 for (DataType tf_dtype : dtypes) { in ConvertDataTypesToUnrankedTensorTypes() local
|