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