| /external/tensorflow/tensorflow/core/common_runtime/ |
| rendezvous_util.h | 20 #include "tensorflow/core/framework/rendezvous.h"
28 // Uses `rendezvous` to send tensors in `tensors_to_send`. `device_context`
34 Rendezvous* rendezvous, DeviceContext* device_context,
38 // Uses `rendezvous` to obtain tensors. `device_context` should be the
43 Rendezvous* rendezvous, DeviceContext* device_context,
48 Status RecvOutputsFromRendezvous(Rendezvous* rendezvous, NamedTensors* out,
49 const Rendezvous::Args& args)
[all...] |
| rendezvous_util.cc | 23 Rendezvous* rendezvous, DeviceContext* device_context,
38 if (!rendezvous) {
39 return errors::InvalidArgument("Rendezvous is null.");
42 Rendezvous::ParsedKey parsed;
44 Rendezvous::Args rendez_args;
49 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(keys[i], &parsed));
51 rendezvous->Send(parsed, rendez_args, tensors_to_send[i], false));
57 Rendezvous* rendezvous, DeviceContext* device_context
[all...] |
| executor.h | 20 #include "tensorflow/core/framework/rendezvous.h"
40 // Rendezvous* rendezvous = NewNaiveRendezvous();
41 // TF_CHECK_OK(rendezvous->Send("input", some_input_tensor));
42 // TF_CHECK_OK(executor->Run({ExecutorOpts, rendezvous, nullptr}));
43 // TF_CHECK_OK(rendezvous->Recv("output", &output_tensor));
56 // ensure objects passed in Args (rendezvous, stats_collector, etc.)
65 // RunAsync() uses the given "rendezvous", if not null, as the
86 Rendezvous* rendezvous = nullptr member in struct:tensorflow::Executor::Args
[all...] |
| kernel_benchmark_testlib.cc | 47 Rendezvous* rendez, const char* executor_type) {
94 args.rendezvous = rendez_;
127 return Rendezvous::CreateKey(send_device, send_device_incarnation,
137 // Gets inputs' and outputs' rendezvous keys.
153 args.rendezvous = rendez_;
160 Rendezvous::ParsedKey parsed;
161 TF_CHECK_OK(Rendezvous::ParseKey(p.first, &parsed));
162 TF_CHECK_OK(rendez_->Send(parsed, Rendezvous::Args(), p.second, false));
166 Rendezvous::ParsedKey parsed;
167 TF_CHECK_OK(Rendezvous::ParseKey(key, &parsed))
[all...] |
| process_function_library_runtime.h | 51 // `rendezvous`. `key_prefix` is used as a prefix for the keys sent to the
52 // Rendezvous. `device_context` should be the DeviceContext of the device
62 Rendezvous* rendezvous);
65 // `source_device`) using `rendezvous`. Uses `key_prefix` to construct the
75 Rendezvous* rendezvous, std::vector<Tensor>* received_tensors,
|
| function.cc | 484 opts.rendezvous = ctx->rendezvous();
955 Rendezvous* rendezvous = opts.rendezvous; local
1054 Rendezvous* rendezvous = new IntraProcessRendezvous(device_mgr_); local
1130 Rendezvous* rendezvous = new IntraProcessRendezvous(device_mgr_); local
[all...] |
| process_function_library_runtime_test.cc | 271 opts.rendezvous = rendezvous_;
287 opts.rendezvous = rendezvous_;
304 opts.rendezvous = rendezvous_;
322 opts.rendezvous = rendezvous_;
342 opts.rendezvous = rendezvous_;
364 opts.rendezvous = rendezvous_;
392 opts.rendezvous = rendezvous_;
|
| process_function_library_runtime.cc | 115 Rendezvous* rendezvous) {
119 string key = Rendezvous::CreateKey(source_device, src_incarnation,
124 rendezvous, device_context, alloc_attrs, keys, tensors_to_send));
133 const std::vector<AllocatorAttributes>& alloc_attrs, Rendezvous* rendezvous,
138 string key = Rendezvous::CreateKey(source_device, src_incarnation,
142 RecvOutputsFromRendezvousAsync(rendezvous, device_context, alloc_attrs, keys,
742 // creates rendezvous, etc.
744 // component function will get a separate rendezvous created by its FLR
948 auto rendezvous = opts.rendezvous; local
[all...] |
| graph_runner.cc | 45 // A simple rendezvous class.
48 class SimpleRendezvous : public Rendezvous {
135 string full_key = Rendezvous::CreateKey("/device:CPU:0", 1, "/device:CPU:1",
137 Rendezvous::ParsedKey parsed;
138 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(full_key, &parsed));
139 TF_RETURN_IF_ERROR(rendez->Send(parsed, Rendezvous::Args(), in.second,
149 // Create the local executor and the Rendezvous for fetching back the
179 args.rendezvous = rendez;
193 Rendezvous::CreateKey("/device:CPU:0", 1, "/device:CPU:1",
195 Rendezvous::ParsedKey parsed
[all...] |
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| delegate_data.cc | 40 // Note that Rendezvous is ref-counted so it will be automatically deleted.
41 tensorflow::Rendezvous* rendezvous = local
46 /*async=*/false, std::move(device_mgr), rendezvous));
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| graph_mgr.cc | 340 Rendezvous* rendezvous = worker_env_->rendezvous_mgr->Find(step_id); local
350 SendTensorsToRendezvous(rendezvous, nullptr, {}, keys, tensors_to_send);
351 rendezvous->Unref();
356 Rendezvous* rendezvous = worker_env_->rendezvous_mgr->Find(step_id); local
357 Status s = RecvOutputsFromRendezvous(rendezvous, out, Rendezvous::Args());
358 rendezvous->Unref();
370 Rendezvous* rendezvous = worker_env_->rendezvous_mgr->Find(step_id) local
426 RemoteRendezvous* rendezvous = worker_env_->rendezvous_mgr->Find(step_id); local
[all...] |
| graph_mgr.h | 166 Rendezvous* rendezvous,
|
| /external/autotest/client/tests/netpipe/ |
| netpipe.py | 55 self.job.barrier(server_tag, 'start', 600).rendezvous(*all)
59 self.job.barrier(server_tag, 'stop', 300).rendezvous(*all)
62 self.job.barrier(client_tag, 'start', 600).rendezvous(*all)
67 self.job.barrier(client_tag, 'stop', 300).rendezvous(*all)
|
| /external/autotest/client/tests/barriertest/ |
| barriertest.py | 16 # Basic barrier rendezvous test.
18 listen_server=server).rendezvous(*hostnames)
19 logging.info('1. rendezvous "First" complete.')
29 # A regular rendezvous, this time testing the abort functionality.
33 ).rendezvous(abort=True, *hostnames)
42 raise error.TestFail('Explicit barrier rendezvous abort failed.')
43 logging.info('3. rendezvous(abort=True) complete.')
66 raise error.TestFail('Explicit barrier rendezvous abort failed.')
|
| /external/tensorflow/tensorflow/core/kernels/ |
| sendrecv_ops.cc | 48 // to formulate the unique rendezvous key.
69 // proactively cache the rendezvous key for the top-level.
71 OP_REQUIRES_OK(ctx, Rendezvous::ParseKey(parsed_key_.buf_, &parsed_key_));
79 ctx, ctx->rendezvous() != nullptr,
80 errors::Internal("Op kernel context needs to provide a rendezvous."));
86 Rendezvous::Args args;
92 // Use the cached rendezvous key.
94 ctx->SetStatus(ctx->rendezvous()->Send(parsed_key_, args, ctx->input(0),
98 Rendezvous::ParsedKey in_loop_parsed;
102 Rendezvous::ParseKey(in_loop_parsed.buf_, &in_loop_parsed))
[all...] |
| /external/tensorflow/tensorflow/c/eager/ |
| c_api_internal.h | 48 #include "tensorflow/core/framework/rendezvous.h"
69 tensorflow::Rendezvous* rendezvous)
73 async, device_mgr, device_mgr_owned, rendezvous) {}
|
| /external/autotest/client/tests/profiler_sync/ |
| profiler_sync.py | 38 b2.rendezvous(*all_ids)
|
| /external/autotest/client/tests/netperf2/ |
| netperf2.py | 92 600).rendezvous(*all)
96 test_time+300).rendezvous(*all)
103 600).rendezvous(*all)
108 300).rendezvous(*all)
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| kernel_and_device.cc | 241 params.rendezvous = rendez_;
310 // We don't pass rendezvous from eager context because we can get tensor
316 Rendezvous* rendezvous = new IntraProcessRendezvous(pflr_->device_mgr()); local
317 opts.rendezvous = rendezvous;
352 rendezvous->Unref();
|
| context.h | 41 #include "tensorflow/core/framework/rendezvous.h"
82 Rendezvous* rendezvous);
87 Rendezvous* rendezvous);
160 Rendezvous* GetRendezvous() const { return rendezvous_; }
222 const gtl::FlatMap<string, uint64>& remote_contexts, Rendezvous* r,
268 Rendezvous* rendezvous_;
|
| context.cc | 50 Rendezvous* rendezvous)
52 /*device_mgr_owned*/ true, rendezvous) {}
57 bool device_mgr_owned, Rendezvous* rendezvous)
60 rendezvous_(rendezvous),
463 const gtl::FlatMap<string, uint64>& remote_contexts, Rendezvous* r,
|
| /external/grpc-grpc/src/python/grpcio/grpc/ |
| _channel.py | 453 rendezvous = _Rendezvous(state, None, None, deadline)
454 return deadline, None, rendezvous
462 rendezvous = _Rendezvous(state, call, None, deadline)
463 return state.response, rendezvous
499 deadline, serialized_request, rendezvous = _start_unary_request(
502 return None, None, None, rendezvous
516 state, operations, deadline, rendezvous = self._prepare(
519 raise rendezvous
540 state, operations, deadline, rendezvous = self._prepare(
543 raise rendezvous
[all...] |
| /external/autotest/client/common_lib/ |
| barrier_unittest.py | 98 # Basic rendezvous testing
103 # The rendezvous should time out here and throw a
116 # The rendezvous should abort here and throw a
124 # The rendezvous should time out here and throw a
131 # The rendezvous should time out here and throw a
145 # The rendezvous should abort here and throw a
165 b1.rendezvous('127.0.0.1#0', '127.0.0.1#1', abort=abort)
167 b1.rendezvous('127.0.0.1#0', '127.0.0.1#1')
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| map_defun_op.cc | 254 opts->rendezvous = ctx->rendezvous();
|
| single_threaded_executor.cc | 209 params.rendezvous = args.rendezvous;
|
