Home
last modified time | relevance | path

Searched refs:edge_name (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.cc51 Allocator* out_allocator, StringPiece edge_name, in CopyHostToDevice() argument
109 Allocator* out_allocator, StringPiece edge_name, in CopyDeviceToHost() argument
122 [edge_name, src, send_dev_context, out_allocator, status_cb]( in CopyDeviceToHost()
137 send_dev_context->CopyDeviceTensorToCPU(&from, edge_name, src, to, in CopyDeviceToHost()
161 send_dev_context->CopyDeviceTensorToCPU(input, edge_name, src, output, in CopyDeviceToHost()
234 void CopyTensor::ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context, in ViaDMA() argument
240 port::Tracing::ScopedAnnotation annotation(edge_name); in ViaDMA()
241 VLOG(1) << "Copy " << edge_name; in ViaDMA()
292 out_allocator, edge_name, dst, output](StatusCallback delete_and_done_, in ViaDMA()
299 CopyHostToDevice(cpu_tensor, cpu_allocator, out_allocator, edge_name, in ViaDMA()
[all …]
Dgraph_runner.cc65 string edge_name = parsed.edge_name.ToString(); in Send() local
66 if (table_.count(edge_name) > 0) { in Send()
69 table_[edge_name] = val; in Send()
78 string key = parsed.edge_name.ToString(); in RecvAsync()
Dcopy_tensor.h45 static void ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context,
Dgpu_device_context.h59 void CopyDeviceTensorToCPU(const Tensor* device_tensor, StringPiece edge_name,
Drendezvous_mgr.cc110 CopyTensor::ViaDMA(parsed.edge_name, send_args.device_context, in SameWorkerRecvDone()
/external/tensorflow/tensorflow/core/framework/
Drendezvous.cc46 edge_name = StringPiece(buf_.data() + (b.edge_name.data() - b_base), in operator =()
47 b.edge_name.size()); in operator =()
109 out->edge_name = StringPiece(parts[3].data(), parts[3].size()); in ParseKey()
Drendezvous.h68 StringPiece edge_name; member
/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_device_context.h38 void CopyDeviceTensorToCPU(const Tensor *device_tensor, StringPiece edge_name,
Dsycl_device_context.cc103 StringPiece edge_name, in CopyDeviceTensorToCPU() argument
/external/v8/src/profiler/
Dheap-snapshot-generator.cc88 const char* prefix, const char* edge_name, int max_depth, int indent) { in Print() argument
91 prefix, edge_name); in Print()
111 const char* edge_name = index.start(); in Print() local
115 edge_name = edge.name(); in Print()
122 edge_name = edge.name(); in Print()
125 edge_name = edge.name(); in Print()
133 edge_name = edge.name(); in Print()
137 edge_name = edge.name(); in Print()
142 edge.to()->Print(edge_prefix, edge_name, max_depth, indent + 2); in Print()
Dheap-snapshot-generator.h129 const char* prefix, const char* edge_name, int max_depth, int indent);
/external/tensorflow/tensorflow/core/distributed_runtime/
Dbase_rendezvous_mgr.cc281 CopyTensor::ViaDMA(parsed.edge_name, send_args.device_context, in SameWorkerRecvDone()
/external/autotest/client/site_tests/firmware_TouchMTB/
Dvalidators.py723 edge_name = direction.split('_')[-1]
724 metric_name = self.mnprops.RANGE.format(edge_name)