Searched refs:replace_shape_ptr (Results 1 – 3 of 3) sorted by relevance
94 buffers_.replace_shape_ptr(on_device_shape_); in set_buffers()106 buffers_.replace_shape_ptr(on_device_shape_); in set_shapes()
53 buffers_.replace_shape_ptr(on_device_shape_); in ShapedBuffer()64 buffers_.replace_shape_ptr(on_device_shape_); in operator =()
135 void replace_shape_ptr(const Shape& shape) { in replace_shape_ptr() function