Home
last modified time | relevance | path

Searched refs:input_shape_values_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_shape_optimization_profiles.h211 input_shape_values_.push_back(actual_shape_values_); in AddShape()
287 std::vector<std::vector<nvinfer1::Dims>> input_shape_values_; variable
Dtrt_shape_optimization_profiles.cc275 dimvec.insert(dimvec.end(), input_shape_values_[i].begin(), in InitProfiles()
276 input_shape_values_[i].end()); in InitProfiles()