Searched refs:remote_shape (Results 1 – 1 of 1) sorted by relevance
118 void SetRemoteShape(std::unique_ptr<TensorShape> remote_shape) { in SetRemoteShape() argument119 remote_shape_ = std::move(remote_shape); in SetRemoteShape()