Home
last modified time | relevance | path

Searched refs:SetRemoteShapeAndDevice (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_execute_node.cc103 Status s = retvals[i]->SetRemoteShapeAndDevice( in RunAsync()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.h210 Status SetRemoteShapeAndDevice(const TensorShape& shape, const Device* d,
Dtensor_handle.cc819 return SetRemoteShapeAndDevice(shape, d, context_view_id, /*op_device=*/""); in SetRemoteShape()
822 Status TensorHandle::SetRemoteShapeAndDevice(const TensorShape& shape, in SetRemoteShapeAndDevice() function in tensorflow::TensorHandle
Dtensor_handle_test.cc398 TF_ASSERT_OK(h->SetRemoteShapeAndDevice( in TEST_F()