Home
last modified time | relevance | path

Searched refs:TransposeTensor (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1154 Status Converter::TransposeTensor(nvinfer1::ITensor* input_tensor, in TransposeTensor() function in tensorflow::tensorrt::convert::Converter
1666 TF_RETURN_IF_ERROR(params->converter->TransposeTensor( in BinaryTensorOpWeight()
1752 TF_RETURN_IF_ERROR(params->converter->TransposeTensor( in BinaryTensorOpWeight()
1829 TF_RETURN_IF_ERROR(params->converter->TransposeTensor( in ConvertConv2DHelper()
1926 TF_RETURN_IF_ERROR(params->converter->TransposeTensor( in ConvertConv2DHelper()
2072 params->converter->TransposeTensor(input_tensor, perm, &output_tensor)); in ConvertTranspose()
2444 TF_RETURN_IF_ERROR(params->converter->TransposeTensor( in ConvertStridedSliceHelper()
2458 TF_RETURN_IF_ERROR(params->converter->TransposeTensor( in ConvertStridedSliceHelper()
2669 TF_RETURN_IF_ERROR(params->converter->TransposeTensor( in ConvertPool()
2722 TF_RETURN_IF_ERROR(params->converter->TransposeTensor( in ConvertPool()
[all …]
Dconvert_nodes.h491 Status TransposeTensor(nvinfer1::ITensor* input_tensor,
Dconvert_nodes_test.cc691 TEST_F(ConverterTest, TransposeTensor) { in TEST_F() argument
698 converter_->TransposeTensor(input_tensor, {0, 1}, &output_tensor), in TEST_F()
704 converter_->TransposeTensor(input_tensor, {1, 0, 2, 3}, &output_tensor), in TEST_F()
709 converter_->TransposeTensor(input_tensor, {0, 3, 1, 2}, &output_tensor)); in TEST_F()