Searched defs:tf_type (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | utils.cc | 87 string DebugString(const DataType tf_type) { in DebugString()
206 Status TfTypeToTrtType(DataType tf_type, nvinfer1::DataType* trt_type) { in TfTypeToTrtType()
224 Status TrtTypeToTfType(nvinfer1::DataType trt_type, DataType* tf_type) { in TrtTypeToTfType()
|
| D | convert_nodes_test.cc | 1412 DataType tf_type) { in AsTensor()
1444 Tensor ConstructTensor(int data_size, const T& value, DataType tf_type) { in ConstructTensor()
1455 DataType tf_type; in CheckDataTypeMatches() local
1535 DataType tf_type; in AddTestTensorWithTFDims() local
1602 const std::vector<T>& values, DataType tf_type) { in AddTestWeights()
1819 DataType tf_type, const std::vector<T>& values, in AddTestTensor()
2127 NodeDef CreateFusedBatchNormOp(DataType tf_type, std::string data_format, in CreateFusedBatchNormOp()
2713 DataType tf_type) -> NodeDef { in TEST_P()
3133 bool pad_per_class = false) -> NodeDef { in TEST_P()
3311 NodeDef CreateUnaryOp(DataType tf_type) { in CreateUnaryOp()
[all …]
|
| D | convert_nodes.cc | 2098 Status GetNodeDefTfType(const NodeDef& node_def, DataType* tf_type, in GetNodeDefTfType()
2109 Status GetInputTfType(const OpConverterParams& params, DataType* tf_type, in GetInputTfType()
2121 Status GetOutputTfType(const OpConverterParams& params, DataType* tf_type) { in GetOutputTfType()
2129 DataType tf_type; in AllowDataTypes() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | string_to_number_op_test.py | 32 def _test(self, tf_type, good_pairs, bad_pairs): argument
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | util.py | 134 def get_tf_type_name(tf_type): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | check_ops.py | 1517 def assert_type_v2(tensor, tf_type, message=None, name=None): argument
1538 def assert_type(tensor, tf_type, message=None, name=None): argument
|