Searched defs:InputBuffers (Results 1 – 5 of 5) sorted by relevance
41 struct InputBuffers { struct42 ~InputBuffers() { in ~InputBuffers()47 std::deque<InputBuffer*> free_buffers;
161 struct InputBuffers { struct162 explicit InputBuffers(xla::Shape device_shape) in InputBuffers() argument170 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;
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
310 struct InputBuffers : public std::unordered_map<Node*, InputBuffer> { struct311 InputBuffer& lookup(Node* function) { in lookup()
218 class InputBuffers { class