Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/kernels/
Dinfeed_ops.cc75 xla::StatusOr<Tensor> TransposeTensor(OpKernelContext* ctx, in TransposeTensor() function
207 TransposeTensor(ctx, input_tensor, shape)); in AutoTransposeAndLinearize()
437 TransposeTensor(ctx, input_tensor, xla_shape_)); in DoWork()
501 TransposeTensor(ctx, *tensor, tuple_shape_.tuple_shapes(i))); in DoWork()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1580 Status Converter::TransposeTensor(nvinfer1::ITensor* input_tensor, in TransposeTensor() function in tensorflow::tensorrt::convert::Converter
2343 TF_RETURN_IF_ERROR(params->converter->TransposeTensor( in ConvertConv2DHelper()
2468 TF_RETURN_IF_ERROR(params->converter->TransposeTensor( in ConvertConv2DHelper()
2526 TF_RETURN_IF_ERROR(params->converter->TransposeTensor( in ConvertTranspose()
3004 TF_RETURN_IF_ERROR(params->converter->TransposeTensor( in ConvertStridedSliceHelper()
3018 params->converter->TransposeTensor(tensor, inv_transpose_order, &tensor, in ConvertStridedSliceHelper()
3337 TF_RETURN_IF_ERROR(params->converter->TransposeTensor( in ConvertConv3DHelper()
3401 TF_RETURN_IF_ERROR(params->converter->TransposeTensor( in ConvertConv3DHelper()
3470 TF_RETURN_IF_ERROR(params->converter->TransposeTensor( in ConvertPool3D()
3497 TF_RETURN_IF_ERROR(params->converter->TransposeTensor( in ConvertPool3D()
[all …]
Dconvert_nodes.h527 Status TransposeTensor(nvinfer1::ITensor* input_tensor,
Dconvert_nodes_test.cc910 TEST_F(ConverterTest, TransposeTensor) { in TEST_F() argument
917 converter_->TransposeTensor(input_tensor, {0, 1}, &output_tensor, in TEST_F()
924 converter_->TransposeTensor(input_tensor, {1, 0, 2, 3}, &output_tensor, in TEST_F()
929 TF_EXPECT_OK(converter_->TransposeTensor( in TEST_F()