| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | utils.cc | 87 string DebugString(const DataType tf_type) { in DebugString() argument
88 switch (tf_type) { in DebugString()
206 Status TfTypeToTrtType(DataType tf_type, nvinfer1::DataType* trt_type) { in TfTypeToTrtType() argument
207 switch (tf_type) { in TfTypeToTrtType()
219 DataTypeString(tf_type)); in TfTypeToTrtType()
224 Status TrtTypeToTfType(nvinfer1::DataType trt_type, DataType* tf_type) { in TrtTypeToTfType() argument
227 *tf_type = DT_FLOAT; in TrtTypeToTfType()
230 *tf_type = DT_HALF; in TrtTypeToTfType()
233 *tf_type = DT_INT32; in TrtTypeToTfType()
|
| D | convert_nodes_test.cc | 1412 DataType tf_type) { in AsTensor() argument
1413 Tensor ret(tensor_buffer_allocator_.get(), tf_type, in AsTensor()
1415 if (tf_type == DT_FLOAT) { in AsTensor()
1418 } else if (tf_type == DT_HALF) { in AsTensor()
1422 } else if (tf_type == DT_INT32) { in AsTensor()
1427 << DataTypeString(tf_type); in AsTensor()
1444 Tensor ConstructTensor(int data_size, const T& value, DataType tf_type) { in ConstructTensor() argument
1446 return AsTensor<T>(values, {data_size}, tf_type); in ConstructTensor()
1455 DataType tf_type; in CheckDataTypeMatches() local
1456 TF_ASSERT_OK(TrtTypeToTfType(trt_dtype, &tf_type)); in CheckDataTypeMatches()
[all …]
|
| D | utils.h | 107 string DebugString(const DataType tf_type);
171 Status TfTypeToTrtType(DataType tf_type, nvinfer1::DataType* trt_type);
172 Status TrtTypeToTfType(nvinfer1::DataType trt_type, DataType* tf_type);
|
| D | convert_nodes.cc | 879 Status TRT_TensorOrWeights::GetTfType(DataType* tf_type) const { in GetTfType()
882 return TrtTypeToTfType(trt_type, tf_type); in GetTfType()
886 *tf_type = weights().GetTensor().dtype(); in GetTfType()
2098 Status GetNodeDefTfType(const NodeDef& node_def, DataType* tf_type, in GetNodeDefTfType() argument
2105 *tf_type = attrs.get<DataType>(type_attr_name); in GetNodeDefTfType()
2109 Status GetInputTfType(const OpConverterParams& params, DataType* tf_type, in GetInputTfType() argument
2116 return inputs[pos].GetTfType(tf_type); in GetInputTfType()
2121 Status GetOutputTfType(const OpConverterParams& params, DataType* tf_type) { in GetOutputTfType() argument
2122 return GetNodeDefTfType(params.node_def, tf_type, kOutputTypeAttrName); in GetOutputTfType()
2129 DataType tf_type; in AllowDataTypes() local
[all …]
|
| D | convert_nodes.h | 304 Status GetTfType(DataType* tf_type) const;
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | string_to_number_op_test.py | 32 def _test(self, tf_type, good_pairs, bad_pairs): argument
37 input_string, out_type=tf_type)
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | util.py | 126 tf_type = _MAP_TFLITE_ENUM_TO_TF_TYPES.get(tflite_enum_type)
127 if tf_type is None:
131 return tf_type
134 def get_tf_type_name(tf_type): argument
136 return "tf." + tf_type.name if tf_type else None
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | check_ops.py | 1517 def assert_type_v2(tensor, tf_type, message=None, name=None): argument
1532 assert_type(tensor=tensor, tf_type=tf_type, message=message, name=name)
1538 def assert_type(tensor, tf_type, message=None, name=None): argument
1555 tf_type = dtypes.as_dtype(tf_type)
1559 if tensor.dtype != tf_type:
1561 raise TypeError('%s tensor must be of type %s' % (message, tf_type))
1565 (message, tensor.name if hasattr(tensor, 'name') else '', tf_type))
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.debugging.pbtxt | 93 …argspec: "args=[\'tensor\', \'tf_type\', \'message\', \'name\'], varargs=None, keywords=None, defa…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.debugging.pbtxt | 93 …argspec: "args=[\'tensor\', \'tf_type\', \'message\', \'name\'], varargs=None, keywords=None, defa…
|
| D | tensorflow.pbtxt | 881 …argspec: "args=[\'tensor\', \'tf_type\', \'message\', \'name\'], varargs=None, keywords=None, defa…
|