Searched refs:remote_shape_ (Results 1 – 2 of 2) sorted by relevance
177 CHECK(remote_shape_ != nullptr); in Shape()178 *shape = *(remote_shape_.get()); in Shape()190 *num_dims = remote_shape_->dims(); in NumDims()205 *dim = remote_shape_->dim_size(dim_index); in Dim()220 *num_elements = remote_shape_->num_elements(); in NumElements()
119 remote_shape_ = std::move(remote_shape); in SetRemoteShape()173 std::unique_ptr<TensorShape> remote_shape_; variable