Searched defs:buffer_tree (Results 1 – 5 of 5) sorted by relevance
67 ShapeTree<se::ScopedDeviceMemory<uint8_t>> buffer_tree(literal_shape); in TransferLiteralToInfeed() local
336 se::Stream* stream, const ShapeTree<MaybeOwningDeviceMemory>& buffer_tree) { in WriteRootTupleIndexTable()
138 ShapeTree<MaybeOwningDeviceMemory> buffer_tree( in ExecutionInputsFromScopedShapedBuffers() local
128981 struct rb_root buffer_tree; member