Searched refs:SetRemoteShape (Results 1 – 2 of 2) sorted by relevance
118 void SetRemoteShape(std::unique_ptr<TensorShape> remote_shape) { in SetRemoteShape() function
632 (*result)->SetRemoteShape(MakeUnique<TensorShape>(tensor->shape())); in EagerRemoteSendTensor()753 retvals[i]->SetRemoteShape(MakeUnique<TensorShape>( in EagerRemoteExecute()774 retvals[i]->SetRemoteShape( in EagerRemoteExecute()