• Home
  • Raw
  • Download

Lines Matching refs:std

56     virtual std::tuple<int, std::vector<OutputShape>, Timing> compute(
61 virtual std::tuple<int, int, ExecuteFencedInfoCallback, Timing> computeFenced(
62 const std::vector<int>& waitFor, const OptionalTimePoint& deadline,
78 virtual std::tuple<int, std::vector<OutputShape>, Timing> execute(
79 const std::vector<ModelArgumentInfo>& inputs,
80 const std::vector<ModelArgumentInfo>& outputs,
81 const std::vector<const RuntimeMemory*>& memories, const SharedBurst& burstController,
88 virtual std::tuple<int, int, ExecuteFencedInfoCallback, Timing> executeFenced(
89 const std::vector<ModelArgumentInfo>& inputs,
90 const std::vector<ModelArgumentInfo>& outputs,
91 const std::vector<const RuntimeMemory*>& memories, const std::vector<int>& waitFor,
97 virtual std::pair<int, std::shared_ptr<RuntimeExecution>> createReusableExecution(
98 const std::vector<ModelArgumentInfo>& inputs,
99 const std::vector<ModelArgumentInfo>& outputs,
100 const std::vector<const RuntimeMemory*>& memories, MeasureTiming measure,
108 using ModelFactory = std::function<Model()>;
111 std::vector<SharedHandle> modelCache;
112 std::vector<SharedHandle> dataCache;
115 using CacheDir = std::string;
118 std::variant<CacheDir, CacheHandles> variant;
130 virtual const std::string& getName() const = 0;
131 virtual const std::string& getVersionString() const = 0;
135 virtual const std::vector<Extension>& getSupportedExtensions() const = 0;
138 virtual std::vector<bool> getSupportedOperations(const MetaModel& metaModel) const = 0;
146 virtual std::pair<uint32_t, uint32_t> getNumberOfCacheFilesNeeded() const = 0;
150 virtual std::pair<int, std::shared_ptr<RuntimePreparedModel>> prepareModel(
153 const std::optional<CacheToken>& maybeToken) const = 0;
157 virtual std::pair<int, std::unique_ptr<RuntimeMemory>> allocate(const MemoryDescriptor& desc,
165 const std::vector<std::shared_ptr<Device>>& getDrivers() const { in getDrivers()
196 static std::shared_ptr<Device> getCpuDevice();
201 std::vector<std::shared_ptr<Device>> forTest_getDevices() const { return mDevices; } in forTest_getDevices()
204 void forTest_setDevices(std::vector<std::shared_ptr<Device>> devices) { in forTest_setDevices()
205 mDevices = std::move(devices); in forTest_setDevices()
219 static std::shared_ptr<Device> forTest_makeDriverDevice(const SharedDevice& device);
235 std::vector<std::shared_ptr<Device>> mDevices;
238 std::vector<std::shared_ptr<Device>> mDevicesCpuOnly;
255 std::vector<SharedDevice> getDevices();