Searched refs:input_partial_shapes_ (Results 1 – 1 of 1) sorted by relevance
265 std::vector<PartialTensorShape> input_partial_shapes_; member in tensorflow::tensorrt::TRTEngineOp418 context->GetAttr("input_shapes", &input_partial_shapes_)); in TRTEngineOp()432 input_mask_.resize(input_partial_shapes_.size()); in TRTEngineOp()439 for (int i = 0; i < input_partial_shapes_.size(); i++) { in TRTEngineOp()441 input_partial_shapes_filtered_.push_back(input_partial_shapes_[i]); in TRTEngineOp()498 if (input_partial_shapes_.empty()) { in TRTEngineOp()508 context, !input_partial_shapes_.empty(), in TRTEngineOp()854 cache_res->profiles_.InitProfiles(input_partial_shapes_, in ComputeAsync()1057 : input_partial_shapes_; in BuildEngine()1319 : input_partial_shapes_; in AllocateCalibrationResources()