Searched refs:SetSrcShape (Results 1 – 2 of 2) sorted by relevance
146 void SetSrcShape(const TensorShape& shape) { src_shape_ = shape; } in SetSrcShape() function
383 captured_state_->SetSrcShape(shape); in StartSendPackedHandle()448 captured_state->SetSrcShape(tensor.shape()); in StartRemoteSendTensor()