Searched defs:InputBuffers (Results 1 – 6 of 6) sorted by relevance
161 struct InputBuffers { struct162 explicit InputBuffers(xla::Shape device_shape) in InputBuffers() function170 xla::ShapedBuffer ToShapedBuffer(xla::Shape host_shape, in ToShapedBuffer()184 xla::ShapeTree<xla::MaybeOwningDeviceMemory> buffers;187 std::vector<VariableInfo> variables;191 std::vector<int> variable_index;
40 struct InputBuffers { struct41 ~InputBuffers() { in ~InputBuffers()46 std::deque<InputBuffer*> free_buffers;
344 std::vector<std::unique_ptr<MemoryBuffer>> InputBuffers; in loadAllFilesForIndex() local438 std::vector<std::unique_ptr<MemoryBuffer>> InputBuffers; in thinLink() local551 auto InputBuffers = loadAllFilesForIndex(*Index); in import() local582 auto InputBuffers = loadAllFilesForIndex(*Index); in internalize() local665 std::vector<std::unique_ptr<MemoryBuffer>> InputBuffers; in runAll() local
759 : InputBuffers(componentName, name) { } in InputBuffers() function810 : InputBuffers(componentName, name), in InputBuffers() function840 : InputBuffers(componentName, name), in InputBuffers() function974 : InputBuffers(componentName, name) { } in InputBuffers() function
310 struct InputBuffers : public std::unordered_map<Node*, InputBuffer> { struct311 InputBuffer& lookup(Node* function) { in lookup()
203 class InputBuffers { class