Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc399 const nvinfer1::Dims input_dims = input.GetTrtDims(); in GetTrtBroadcastShape()
432 std::max(operand_l.GetTrtDims().nbDims + in GetTrtBroadcastShape()
434 operand_r.GetTrtDims().nbDims + in GetTrtBroadcastShape()
671 const nvinfer1::Dims dims_0 = inputs.at(0).GetTrtDims(); in VerifyShapesMatch()
673 const nvinfer1::Dims dim_i = inputs.at(i).GetTrtDims(); in VerifyShapesMatch()
871 nvinfer1::Dims TRT_TensorOrWeights::GetTrtDims() const { in GetTrtDims() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
1709 const nvinfer1::Dims input_dims = input.GetTrtDims(); in PrepareTensorForShape()
2540 if (HasStaticShape(inputs.at(0).GetTrtDims())) { in ConvertShape()
2542 nvinfer1::Dims input_dims = inputs.at(0).GetTrtDims(); in ConvertShape()
2593 const nvinfer1::Dims input_nonbatch_dims = input_tensor.GetTrtDims(); in ConvertReshape()
[all …]
Dconvert_nodes.h302 nvinfer1::Dims GetTrtDims() const;
Dconvert_nodes_test.cc489 ExpectTrtDimsEqualsArray({1}, ptr->GetTrtDims()); in TEST()
508 ExpectTrtDimsEqualsArray({1}, ptr->GetTrtDims()); in TEST()
522 ExpectTrtDimsEqualsArray({}, ptr->GetTrtDims()); in TEST()
613 ExpectTrtDimsEqualsArray({non_batch_dim}, output.GetTrtDims()); in TEST_F()