HomeSort by relevance Sort by last modified time
    Searched refs:CopyTensor (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/core/debug/
debug_gateway.h 76 void CopyTensor(const string& node_name, const int output_slot,
debug_gateway.cc 43 CopyTensor(node_name, output_slot, tensor, ctx,
68 void DebugGateway::CopyTensor(const string& node_name, const int output_slot,
  /external/tensorflow/tensorflow/core/common_runtime/
copy_tensor.h 29 class CopyTensor {
66 // CopyTensor::Registration.
copy_tensor.cc 33 RegistrationInfo(DeviceType s, DeviceType r, CopyTensor::CopyFunction cf)
39 CopyTensor::CopyFunction copy_function;
166 void CopyDeviceToDevice(CopyTensor::CopyFunction copy_function,
234 void CopyTensor::ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context,
333 Status CopyTensor::Register(DeviceType sender_device_type,
rendezvous_mgr.cc 80 // checks happen inside CopyTensor::ViaDMA.
105 // Variants are handled by CopyTensor::ViaDMA.
110 CopyTensor::ViaDMA(parsed.edge_name, send_args.device_context,
  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_util.cc 255 static CopyTensor::Registration register_gpu_gpu_copy(
  /external/tensorflow/tensorflow/core/distributed_runtime/
base_rendezvous_mgr.cc 247 // checks happen inside CopyTensor::ViaDMA.
274 // Variants are handled by CopyTensor::ViaDMA.
281 CopyTensor::ViaDMA(parsed.edge_name, send_args.device_context,
  /external/tensorflow/tensorflow/c/eager/
c_api.cc 243 tensorflow::CopyTensor::ViaDMA("copy", src_device_context, dst_device_context,
    [all...]

Completed in 760 milliseconds