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