Home
last modified time | relevance | path

Searched refs:copy_status (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dtridiagonal_solve_op_gpu.cu.cc83 bool copy_status = stream in CopyDeviceToDevice() local
88 if (!copy_status) { in CopyDeviceToDevice()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op.cc454 Status copy_status = batch_util::CopyElementToSlice( in CallFunction() local
456 if (!copy_status.ok()) { in CallFunction()
457 result->UpdateStatus(copy_status, offset); in CallFunction()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc708 Status copy_status; in MakeTensorFromProto() local
709 auto copier = [this, &alloc_attrs, &notifications, &copy_status]( in MakeTensorFromProto()
716 [&n, &copy_status](const Status& s) { in MakeTensorFromProto()
717 if (copy_status.ok()) { in MakeTensorFromProto()
718 copy_status.Update(s); in MakeTensorFromProto()
738 return copy_status; in MakeTensorFromProto()