Searched defs:tf_type (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | utils.cc | 38 string DebugString(const DataType tf_type) { in DebugString()
153 Status TfTypeToTrtType(DataType tf_type, nvinfer1::DataType* trt_type) { in TfTypeToTrtType()
176 Status TrtTypeToTfType(nvinfer1::DataType trt_type, DataType* tf_type) { in TrtTypeToTfType()
|
| D | convert_nodes_test.cc | 1201 const std::vector<int>& input_dims, DataType tf_type) { in AsTensor()
1226 const std::vector<int>& input_dims, DataType tf_type) { in AsTensor()
1240 Tensor ConstructTensor(int data_size, const T& value, DataType tf_type) { in ConstructTensor()
1259 DataType tf_type; in CheckDataTypeMatches() local
1347 DataType tf_type; in AddTestTensorWithTFDims() local
1400 const std::vector<T>& values_inp, DataType tf_type, in AddTestWeights()
1812 DataType tf_type, const std::vector<T>& values_inp, in AddTestTensor()
1988 const DataType tf_type = get_tf_type(); in RunTests() local
2054 const DataType tf_type = get_tf_type(); in RunTests() local
2517 NodeDef CreateFusedBatchNormOp(DataType tf_type, std::string data_format, in CreateFusedBatchNormOp()
[all …]
|
| D | convert_nodes.cc | 1715 Status GetNodeDefTfType(const NodeDef& node_def, DataType* tf_type, in GetNodeDefTfType()
1738 Status GetInputTfType(const OpConverterParams& params, DataType* tf_type, in GetInputTfType()
1748 Status GetOutputTfType(const OpConverterParams& params, DataType* tf_type) { in GetOutputTfType()
1756 DataType tf_type; in AllowDataTypes() local
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_check_ops.py | 23 def assert_type(tensor: ragged_tensor.Ragged, tf_type, message=None, name=None): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/strings_ops/ |
| D | string_to_number_op_test.py | 28 def _test(self, tf_type, good_pairs, bad_pairs): argument
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | util.py | 104 def get_tf_type_name(tf_type): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | check_ops.py | 1584 def assert_type_v2(tensor, tf_type, message=None, name=None): argument
1620 def assert_type(tensor, tf_type, message=None, name=None): argument
|
| /external/tensorflow/tensorflow/core/ir/types/ |
| D | dialect.cc | 55 namespace tf_type { namespace
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels_test.cc | 858 #define TEST_KERNEL_TYPE_CONSTRAINT(tf_type, dtype) \ argument
|
| /external/deqp/external/openglcts/modules/gl/ |
| D | gl4cSparseBufferTests.cpp | 5025 unsigned int tf_type, in TransformFeedbackBufferStorageTestCase()
|