| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 305 ScopedStepContainer step_container( in Run() local
309 f_opts.step_container = &step_container; in Run()
341 ScopedStepContainer step_container( in RunWithBorrowedArgs() local
345 f_opts.step_container = &step_container; in RunWithBorrowedArgs()
377 ScopedStepContainer step_container( in RunInstantiated() local
381 f_opts.step_container = &step_container; in RunInstantiated()
421 ScopedStepContainer* step_container = new ScopedStepContainer( in RunAsync() local
425 f_opts.step_container = step_container; in RunAsync()
445 [this, rets, step_container, c_mgr, frame]( in RunAsync()
453 delete step_container; in RunAsync()
|
| D | single_threaded_executor.cc | 219 params.step_container = args.step_container; in RunAsync()
|
| D | iterator_ops.cc | 879 ScopedStepContainer step_container(opts.step_id, [ctx](const string& name) { in TryInit() local
882 opts.step_container = &step_container; in TryInit()
|
| D | dataset_test_base.cc | 191 params_->step_container = step_container_.get(); in CreateOpKernelContext()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 155 ScopedStepContainer step_container(0, [this](const string& name) { in Run() local
158 return this->Run(&step_container, inputs, outputs, stats, step_stats, in Run()
167 ScopedStepContainer step_container(0, [&devices](const string& name) { in Run() local
172 return this->Run(&step_container, inputs, outputs, stats, step_stats, in Run()
206 Status KernelAndDeviceOp::Run(ScopedStepContainer* step_container, in Run() argument
265 params.step_container = step_container; in Run()
305 ScopedStepContainer* step_container, in Run() argument
325 opts.step_container = step_container; in Run()
|
| D | kernel_and_device.h | 86 virtual Status Run(ScopedStepContainer* step_container,
151 Status Run(ScopedStepContainer* step_container,
212 Status Run(ScopedStepContainer* step_container,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 144 auto* step_container = ctx->step_container(); in GetStack() local
145 if (step_container == nullptr) { in GetStack()
148 TF_RETURN_IF_ERROR(rm->Lookup(step_container->name(), key, stack)); in GetStack()
188 auto* step_container = ctx->step_container(); in Compute() local
189 OP_REQUIRES(ctx, step_container != nullptr, in Compute()
191 OP_REQUIRES_OK(ctx, rm->Create(step_container->name(), key, stack)); in Compute()
|
| D | variable_ops.cc | 103 OP_REQUIRES_OK(context, rm->Create(context->step_container()->name(), in Compute()
149 context->step_container()->name(), var_name_)); in Compute()
|
| D | partitioned_function_ops.cc | 216 run_opts.step_container = ctx->step_container(); in RunFunction()
|
| D | function_ops.cc | 257 opts.step_container = ctx->step_container(); in ComputeAsync()
|
| D | tensor_array_ops.cc | 82 TF_RETURN_IF_ERROR(rm->Lookup(ctx->step_container()->name(), in GetTensorArray()
213 rm->Create(ctx->step_container()->name(), key, tensor_array)); in CreateTensorArray()
309 TF_RETURN_IF_ERROR(rm->Lookup(ctx->step_container()->name(), in CreateTensorArray()
368 ctx->step_container()->name(), key, output_tensor_array, creator); in CreateTensorArray()
|
| D | ops_testutil.h | 173 params_->step_container = step_container_.get(); in RunOpKernel()
|
| D | functional_ops.cc | 124 opts->step_container = ctx->step_container(); in SetRunOptions()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler.h | 60 ScopedStepContainer* step_container) in GraphCompiler() argument
64 step_container_(step_container) {} in GraphCompiler()
|
| D | xla_context.cc | 49 ctx->step_container()->name(), kXlaContextResourceName, &context)); in Get()
|
| D | xla_compiler.cc | 127 auto step_container = absl::make_unique<ScopedStepContainer>( in ExecuteGraph() local
132 step_container->name(), XlaContext::kXlaContextResourceName, in ExecuteGraph()
135 GraphCompiler graph_compiler(device, graph.get(), flib, step_container.get()); in ExecuteGraph()
138 step_container.reset(); in ExecuteGraph()
|
| D | graph_compiler.cc | 309 params->step_container = step_container_; in PartiallySetupParams()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | function_ops.cc | 79 opts.step_container = ctx->step_container(); in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 477 ScopedStepContainer* step_container = new ScopedStepContainer( in StartParallelExecutors() local
483 [this, item, collector, cost_graph, step_container, in StartParallelExecutors()
487 delete step_container; in StartParallelExecutors()
495 args.step_container = step_container; in StartParallelExecutors()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor.h | 94 ScopedStepContainer* step_container = nullptr; member
|
| D | direct_session.cc | 538 args.step_container = &run_state.step_container; in RunInternal()
894 args.step_container = &run_state->step_container; in PRunSetup()
1645 : step_container(step_id, [devices, step_id](const string& name) { in RunState()
|
| D | direct_session.h | 190 ScopedStepContainer step_container; member
|
| D | function.cc | 486 opts.step_container = ctx->step_container(); in ComputeAsync()
939 exec_args->step_container = run_opts.step_container; in ExecutorArgsFromOptions()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 627 ScopedStepContainer* step_container = nullptr; member
1163 ScopedStepContainer* step_container() const { in step_container() function
1164 return params_->step_container; in step_container()
|
| D | function.h | 620 ScopedStepContainer* step_container = nullptr; member
|