Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1225 Status Converter::PrepareTensorForShape(const TRT_TensorOrWeights& input, in PrepareTensorForShape() function in tensorflow::tensorrt::convert::Converter
1974 status = params->converter->PrepareTensorForShape( in BinaryTensorOpTensor()
1977 status = params->converter->PrepareTensorForShape( in BinaryTensorOpTensor()
2170 TF_RETURN_IF_ERROR(params->converter->PrepareTensorForShape( in ConvertReshape()
2223 TF_RETURN_IF_ERROR(params->converter->PrepareTensorForShape( in ConvertExpandDims()
2284 TF_RETURN_IF_ERROR(params->converter->PrepareTensorForShape( in ConvertSqueeze()
2438 TF_RETURN_IF_ERROR(params->converter->PrepareTensorForShape( in ConvertStridedSliceHelper()
2483 TF_RETURN_IF_ERROR(params->converter->PrepareTensorForShape( in ConvertStridedSliceHelper()
3775 TF_RETURN_IF_ERROR(params->converter->PrepareTensorForShape( in ConvertGather()
3809 TF_RETURN_IF_ERROR(params->converter->PrepareTensorForShape( in ConvertMatMulHelper()
[all …]
Dconvert_nodes.h500 Status PrepareTensorForShape(const TRT_TensorOrWeights& input,
Dconvert_nodes_test.cc722 converter_->PrepareTensorForShape(tw, GetTestDims({2, 3, 6}), in TEST_F()
730 TF_EXPECT_OK(converter_->PrepareTensorForShape( in TEST_F()
739 TF_EXPECT_OK(converter_->PrepareTensorForShape( in TEST_F()
755 TF_EXPECT_OK(converter_->PrepareTensorForShape( in TEST_F()