Searched refs:inputDimensions_ (Results 1 – 1 of 1) sorted by relevance
93 std::vector<uint32_t> inputDimensions_; member in Conv2DModel167 inputDimensions_ = std::vector<uint32_t>{batches, height, width, depth_in}; in CreateModel()171 inputSize_ = product(inputDimensions_); in CreateModel()188 .dimensionCount = static_cast<uint32_t>(inputDimensions_.size()), in CreateModel()189 .dimensions = inputDimensions_.data(), in CreateModel()