Searched refs:SetRemoteShape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_mgr_test.cc | 82 handle->SetRemoteShape(shape, remote_device_, ctx_->GetContextViewId())); in TEST_F()
128 ->SetRemoteShape(TensorShape({0}), remote_device_, in TEST_F()
142 TF_ASSERT_OK(handle->SetRemoteShape(TensorShape({0}), remote_device_, in TEST_F()
152 TF_ASSERT_OK(handle->SetRemoteShape(TensorShape({0}), remote_device_, in TEST_F()
|
| D | remote_copy_node.cc | 259 Status status = captured_state->dst()->SetRemoteShape( in RunRemoteRecv()
411 Status status = captured_state->dst()->SetRemoteShape( in StartSendPackedHandle()
464 Status status = captured_state->dst()->SetRemoteShape( in StartRemoteSendTensor()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle_test.cc | 208 TF_ASSERT_OK(h2->SetRemoteShape(shape, ListDevices().at(2), in TEST_F()
211 TF_ASSERT_OK(h3->SetRemoteShape(shape, ListDevices().at(3), in TEST_F()
|
| D | tensor_handle.h | 206 Status SetRemoteShape(const TensorShape& shape, const Device* d,
|
| D | execute.cc | 1719 retvals[i]->SetRemoteShape(absl::get<TensorShape>(ret), in GetKernelOutputs()
1747 TF_RETURN_IF_ERROR(retvals[i]->SetRemoteShape( in GetKernelOutputs()
|
| D | tensor_handle.cc | 817 Status TensorHandle::SetRemoteShape(const TensorShape& shape, const Device* d, in SetRemoteShape() function in tensorflow::TensorHandle
|