Searched defs:InputBuffers (Results 1 – 4 of 4) sorted by relevance
40 struct InputBuffers { struct41 ~InputBuffers() { in ~InputBuffers()46 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()185 xla::ShapeTree<xla::MaybeOwningDeviceMemory> buffers;188 std::vector<VariableInfo> variables;192 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
203 class InputBuffers { class