Searched defs:InputBuffers (Results 1 – 6 of 6) 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;
498 std::vector<std::unique_ptr<MemoryBuffer>> InputBuffers; in loadAllFilesForIndex() local604 std::vector<std::unique_ptr<MemoryBuffer>> InputBuffers; in thinLink() local730 auto InputBuffers = loadAllFilesForIndex(*Index); in import() local763 auto InputBuffers = loadAllFilesForIndex(*Index); in internalize() local818 std::vector<std::unique_ptr<MemoryBuffer>> InputBuffers; in codegen() local856 std::vector<std::unique_ptr<MemoryBuffer>> InputBuffers; in runAll() local
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
779 SmallVector<std::unique_ptr<MemoryBuffer>, 8u> InputBuffers; in BundleFiles() local