Home
last modified time | relevance | path

Searched refs:TFE_TensorHandleDeviceName (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_testlib.cc67 const char* device = TFE_TensorHandleDeviceName(handle_, status); in Destroy()
84 const char* device = TFE_TensorHandleDeviceName(handle_, status); in Read()
106 const char* device = TFE_TensorHandleDeviceName(handle_, status); in GeneralAssignment()
168 const char* device = TFE_TensorHandleDeviceName(input, status); in ExtractPerDeviceValues()
191 const char* first_device = TFE_TensorHandleDeviceName(first, status); in Multiply()
Dparallel_device.cc304 TFE_TensorHandleDeviceName(input, status); in ParallelDeviceExecute()
Dparallel_device_test.cc389 const char* device = TFE_TensorHandleDeviceName(input, status); in CollectiveSum()
/external/tensorflow/tensorflow/go/
Dtensor_handle.go116 name := C.TFE_TensorHandleDeviceName(th.c, status.c)
/external/tensorflow/tensorflow/c/eager/
Dcustom_device_testutil.cc119 const char* input_device = TFE_TensorHandleDeviceName(input, s); in LoggingDeviceExecute()
Dc_api.h160 TF_CAPI_EXPORT extern const char* TFE_TensorHandleDeviceName(
Dc_api_distributed_test.cc184 EXPECT_EQ(TFE_TensorHandleDeviceName(packed_handle, status), in TestFunctionWithPackedInput()
Dcustom_device_test.cc245 tensorflow::string(TFE_TensorHandleDeviceName(var_value, status.get()))); in TEST()
Dc_api_test.cc521 const char* device_name = TFE_TensorHandleDeviceName(h, status.get()); in TEST()
557 const char* device_name = TFE_TensorHandleDeviceName(hcpu, status.get()); in TEST()
582 device_name = TFE_TensorHandleDeviceName(retvals[0], status.get()); in TEST()
Dc_api.cc322 const char* TFE_TensorHandleDeviceName(TFE_TensorHandle* h, TF_Status* status) { in TFE_TensorHandleDeviceName() function
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc651 const char* device = TFE_TensorHandleDeviceName(self->handle, &self->status); in EagerTensor_device()