| /external/tensorflow/tensorflow/contrib/verbs/ |
| rdma.h | 107 // type|name_size|name|step_id|request_index|remote_addr/checksum|rkey|...
239 RdmaTensorRequest(uint32_t index, const string& key, int64 step_id,
432 const string& key, int64 step_id, Device* dst_dev,
|
| rdma.cc | 669 const string& key, int64 step_id, Device* dst_dev,
677 RdmaTensorRequest request(request_index, key, step_id, this, dst_dev,
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| resource_mgr.h | 88 // step_id: the unique ID of this step. Doesn't have to be sequential, just
91 ScopedStepContainer(const int64 step_id,
93 : name_(strings::StrCat("__per_step_", step_id)), cleanup_(cleanup) {}
|
| op_kernel.h | 483 int64 step_id = 0; member in struct:tensorflow::OpKernelContext::Params
577 int64 step_id() const { return params_->step_id; } function in class:tensorflow::OpKernelContext
[all...] |
| function.h | 477 int64 step_id = 0; member in struct:tensorflow::FunctionLibraryRuntime::Options
|
| op_kernel.cc | 642 params_->step_id, new_tensor);
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| direct_session.cc | 472 args.step_id = step_id_counter_.fetch_add(1);
482 run_options.debug_options(), args.step_id, executor_step_count,
510 RunState run_state(args.step_id, &devices_);
533 LogMemory::RecordStep(args.step_id, run_state_args.handle);
740 args.step_id = step_id_counter_.fetch_add(1);
742 new RunState(input_names, output_names, args.step_id, &devices_);
775 LogMemory::RecordStep(args.step_id, run_state_args.handle);
[all...] |
| direct_session.h | 175 RunState(int64 step_id, const std::vector<Device*>* devices);
178 const std::vector<string>& pending_output_names, int64 step_id,
|
| function.cc | 298 opts.step_id = ctx->step_id();
776 // Inherit the step_id from the caller.
777 exec_args->step_id = run_opts.step_id;
862 // Inherit the step_id from the caller.
863 exec_args->step_id = run_opts.step_id;
[all...] |
| graph_runner.cc | 180 args.step_id = LogMemory::CONSTANT_FOLDING_STEP_ID;
|
| executor.cc | [all...] |
| /external/tensorflow/tensorflow/contrib/nccl/kernels/ |
| nccl_ops.cc | 47 return strings::StrCat(collective_prefix_, ";", c->step_id(), ";",
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| gpu_event_mgr.cc | 69 ue->bufrec.step_id, ue->bufrec.buf,
|
| gpu_device.cc | 99 step_id_ = context->step_id();
[all...] |
| /external/tensorflow/tensorflow/contrib/data/kernels/ |
| prefetching_kernels.cc | 174 opts.step_id = -std::abs(static_cast<int64>(random::New64()));
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| message_wrappers_test.cc | 88 EXPECT_EQ(13, request.step_id());
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_compiler.cc | 297 int64 step_id) {
308 step_id, [&status, device](const string& name) {
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| iterator_ops.cc | 775 opts.step_id = -std::abs(static_cast<int64>(random::New64()));
776 ScopedStepContainer step_container(opts.step_id, [ctx](const string& name) {
[all...] |
