OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Automated Code Change

Browse Source 
PiperOrigin-RevId: 847190131
This commit is contained in:
A. Unique TensorFlower
2025-12-20 13:47:29 -08:00
committed by TensorFlower Gardener
parent b1082f97e5
commit b1b2723815
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tensorflow/c/eager/custom_device_testutil.cc
View File

@@ -27,8 +27,8 @@ limitations under the License.
namespace {
struct LoggingDevice {
tensorflow::string device_name;
tensorflow::string underlying_device;
std::string device_name;
std::string underlying_device;
// Set to true whenever a TensorHandle is copied onto the device
bool* arrived_flag;
// Set to true whenever an operation is executed
@@ -59,9 +59,10 @@ void LoggedTensorDeallocator(void* data) {
delete reinterpret_cast<LoggedTensor*>(data);
}
TFE_TensorHandle* MakeLoggedTensorHandle(
TFE_Context* context, const tensorflow::string& logging_device_name,
std::unique_ptr<LoggedTensor> t, TF_Status* status) {
TFE_TensorHandle* MakeLoggedTensorHandle(TFE_Context* context,
const std::string& logging_device_name,
std::unique_ptr<LoggedTensor> t,
TF_Status* status) {
auto dtype = TFE_TensorHandleDataType(t->tensor);
TFE_CustomDeviceTensorHandleMethods handle_methods;
handle_methods.num_dims = &LoggedTensorNumDims;