Searched refs:TFE_OpGetDevice (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | custom_device_testutil.cc | 93 const char* requested_placement = TFE_OpGetDevice(original_op, s); in LoggingDeviceExecute()
|
| D | c_api.h | 264 TF_CAPI_EXPORT extern const char* TFE_OpGetDevice(const TFE_Op* op,
|
| D | custom_device_test.cc | 86 ASSERT_EQ(tensorflow::string(TFE_OpGetDevice(reused_op.get(), status.get())), in TEST()
92 ASSERT_EQ(tensorflow::string(TFE_OpGetDevice(reused_op.get(), status.get())), in TEST()
|
| D | c_api_test.cc | 493 const char* device_name = TFE_OpGetDevice(matmul, status); in SetAndGetOpDevices()
498 device_name = TFE_OpGetDevice(matmul, status); in SetAndGetOpDevices()
1758 const char* device = TFE_OpGetDevice(other, status); in CloneOp()
1788 EXPECT_EQ("", std::string(TFE_OpGetDevice(original_var_op, status))); in TEST()
1796 std::string(TFE_OpGetDevice(cloned, status))); in TEST()
1816 EXPECT_EQ("", std::string(TFE_OpGetDevice(cloned_identity, status))); in TEST()
|
| D | c_api.cc | 668 const char* TFE_OpGetDevice(const TFE_Op* op, TF_Status* status) { in TFE_OpGetDevice() function
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device.cc | 279 const char* requested_placement = TFE_OpGetDevice(original_op, status); in ParallelDeviceExecute()
|