Searched refs:GetTrtDims (Results 1 – 11 of 11) sorted by relevance
71 const auto dims = inputs.at(0).GetTrtDims(); in Validate()99 const auto &repl_dims = repl.GetTrtDims(); in Validate()128 const auto dims = tensor.GetTrtDims(); in Convert()169 replics.GetTrtDims()); in Convert()
66 nvinfer1::Dims dims(input.GetTrtDims()); in Convert()90 input.GetTrtDims().nbDims, dims); in Convert()
76 const auto nbDims = dims_input.GetTrtDims().nbDims; in Validate()96 int nbDims = dims_input.GetTrtDims().d[0]; in Convert()184 const auto& dims = input.GetTrtDims(); in Validate()
68 const auto x_dims = x_input_.GetTrtDims(); in Validate()157 x_input_.weights(), x_input_.GetTrtDims()) in Convert()
227 dims = operand.GetTrtDims(); in Initialize()296 TRT_ENSURE(operand.GetTrtDims().nbDims == dims.nbDims); in SetDynamicSize()300 for (int i = 0; i < operand.GetTrtDims().nbDims; i++) { in SetDynamicSize()314 for (int i = 0; i < operand.GetTrtDims().nbDims; i++) { in SetDynamicSize()422 operand.weights().GetTrtWeights(), operand.GetTrtDims()); in ConditionEinsumWeights()894 input_arg.GetTrtDims());
91 DimsAdapter input_dims(params.inputs.at(0).GetTrtDims());
348 input.weights().GetTrtWeights(), input.GetTrtDims()); in get_tensor4TensorOrWeights()
318 const nvinfer1::Dims input_dims = input.GetTrtDims(); in GetTrtBroadcastShape()346 std::max(operand_l.GetTrtDims().nbDims + in GetTrtBroadcastShape()348 operand_r.GetTrtDims().nbDims + in GetTrtBroadcastShape()409 if (p->GetTrtDims().nbDims != broadcasted_dims.NumDims()) { in BroadcastWeights()638 const nvinfer1::Dims dims_0 = inputs.at(0).GetTrtDims(); in VerifyShapesMatch()640 const nvinfer1::Dims dim_i = inputs.at(i).GetTrtDims(); in VerifyShapesMatch()1544 DimsAdapter input_dims(input.GetTrtDims()); in PrepareTensorForShape()1841 if (inputs.at(1).GetTrtDims().nbDims != 4) { in ConvertConv2DHelper()1913 auto weights_shape = inputs.at(1).GetTrtDims(); in ConvertConv2DHelper()2134 DimsAdapter input_dims(inputs.at(0).GetTrtDims()); in ConvertShape()[all …]
147 nvinfer1::Dims TRT_TensorOrWeights::GetTrtDims() const { in GetTrtDims() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights
235 nvinfer1::Dims GetTrtDims() const;
269 EXPECT_THAT(ptr->GetTrtDims(), DimsAreArray({1})); in TEST()288 EXPECT_THAT(ptr->GetTrtDims(), DimsAreArray({1})); in TEST()303 EXPECT_THAT(ptr->GetTrtDims(), DimsAreArray(empty_dims)); in TEST()380 EXPECT_THAT(output.GetTrtDims(), DimsAreArray({non_batch_dim})); in TEST_F()