Searched refs:PrepareTensorForShape (Results 1 – 3 of 3) sorted by relevance
1225 Status Converter::PrepareTensorForShape(const TRT_TensorOrWeights& input, in PrepareTensorForShape() function in tensorflow::tensorrt::convert::Converter1974 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 …]
500 Status PrepareTensorForShape(const TRT_TensorOrWeights& input,
722 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()