Searched refs:input_dim_list_ (Results 1 – 3 of 3) sorted by relevance
46 input_dim_list_.emplace_back(dim); in PluginTensorRT()60 input_dim_list_.emplace_back(dim); in configure()67 sizeof(input_dim_list_.size()) + sizeof(dim.nbDims) + sizeof(dim.d) + in getSerializationSize()80 auto list_size = input_dim_list_.size(); in serialize()81 std::memcpy(buffer, &list_size, sizeof(input_dim_list_.size())); in serialize()82 buffer += sizeof(input_dim_list_.size()); in serialize()84 for (int i = 0; i < input_dim_list_.size(); i++) { in serialize()85 auto dim = input_dim_list_[i]; in serialize()
65 std::vector<nvinfer1::Dims> input_dim_list_; variable
72 for (int i = 0; i < input_dim_list_[0].nbDims; i++) { in enqueue()73 count *= input_dim_list_[0].d[i]; in enqueue()