Searched refs:input_partial_shapes_ (Results 1 – 1 of 1) sorted by relevance
210 std::vector<PartialTensorShape> input_partial_shapes_; member in tensorflow::tensorrt::TRTEngineOp353 context->GetAttr("input_shapes", &input_partial_shapes_)); in TRTEngineOp()416 if (input_partial_shapes_.empty()) { in TRTEngineOp()426 context, !input_partial_shapes_.empty(), in TRTEngineOp()439 input_partial_shapes_, in TRTEngineOp()539 if (input_partial_shapes_.empty()) { in VerifyInputShapes()556 " != ", DebugString(input_partial_shapes_)); in VerifyInputShapes()557 if (input_concrete_shapes.size() != input_partial_shapes_.size()) { in VerifyInputShapes()561 if (input_concrete_shapes[i].dims() != input_partial_shapes_[i].dims()) { in VerifyInputShapes()567 if (input_partial_shapes_[i].dim_size(d) != -1) { in VerifyInputShapes()[all …]