| /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(step_container->Lookup(rm, key, stack)); in GetStack()
187 auto* step_container = ctx->step_container(); in Compute() local
188 OP_REQUIRES(ctx, step_container != nullptr, in Compute()
191 OP_REQUIRES_OK(ctx, step_container->Create(rm, key, stack)); in Compute()
207 ctx->step_container()->MakeResourceHandle<Stack>(key, *ctx->device()); in Compute()
|
| D | partitioned_function_ops.cc | 233 ScopedStepContainer* step_container = new ScopedStepContainer( in RunFunction() local
237 run_opts.step_container = step_container; in RunFunction()
267 step_container](const Status& status) { in RunFunction()
279 delete step_container; in RunFunction()
|
| D | variable_ops.cc | 104 context->step_container()->Create(rm, var_name_, tmp_var)); in Compute()
149 context, context->step_container()->Delete<TemporaryVariableOp::TmpVar>( in Compute()
|
| D | function_ops.cc | 257 opts.step_container = ctx->step_container(); in ComputeAsync()
|
| D | ops_testutil.cc | 161 params_->step_container = step_container_.get(); in RunOpKernel()
|
| D | tensor_array_ops.cc | 84 ctx->step_container()->Lookup(rm, container + ta_handle, tensor_array)); in GetTensorArray()
213 TF_RETURN_IF_ERROR(ctx->step_container()->Create(rm, key, tensor_array)); in CreateTensorArray()
309 TF_RETURN_IF_ERROR(ctx->step_container()->Lookup( in CreateTensorArray()
365 Status s = ctx->step_container()->LookupOrCreate<TensorArray>( in CreateTensorArray()
|
| D | functional_ops.cc | 110 opts->step_container = ctx->step_container(); in SetRunOptions()
|
| D | tensor_array.h | 350 return ctx->step_container()->MakeResourceHandle<TensorArray>( in resource_handle()
|
| D | batch_kernels.cc | 511 opts.step_container = last_task_context->step_container(); in ProcessFuncBatch()
|
| /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 | 48 TF_CHECK_OK(ctx->step_container()->Lookup(ctx->resource_manager(), in Get()
|
| D | xla_compiler.cc | 126 auto step_container = absl::make_unique<ScopedStepContainer>( in ExecuteGraph() local
130 TF_RETURN_IF_ERROR(step_container->Create(device->resource_manager(), in ExecuteGraph()
134 GraphCompiler graph_compiler(device, graph.get(), flib, step_container.get()); in ExecuteGraph()
137 step_container.reset(); in ExecuteGraph()
|
| D | graph_compiler.cc | 332 params->step_container = step_container_; in PartiallySetupParams()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 238 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in Run() argument
275 params.step_container = step_container; in Run()
307 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in Run() argument
346 opts->step_container = step_container; in Run()
|
| D | kernel_and_device.h | 127 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
193 Status Run(ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
272 Status Run(ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 646 ScopedStepContainer step_container( in Run() local
650 f_opts.step_container = &step_container; in Run()
684 ScopedStepContainer step_container( in RunWithBorrowedArgs() local
688 f_opts.step_container = &step_container; in RunWithBorrowedArgs()
723 ScopedStepContainer step_container( in RunInstantiated() local
727 f_opts.step_container = &step_container; in RunInstantiated()
776 ScopedStepContainer* step_container = new ScopedStepContainer( in RunAsync() local
780 f_opts.step_container = step_container; in RunAsync()
797 [this, rets, step_container, raw_cancellation_manager, frame]( in RunAsync()
803 delete step_container; in RunAsync()
|
| D | single_threaded_executor.cc | 254 params.step_container = args.step_container; in Run()
|
| D | iterator_ops.cc | 817 ScopedStepContainer step_container(opts.step_id, [ctx](const string& name) { in TryInit() local
820 opts.step_container = &step_container; in TryInit()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 504 ScopedStepContainer* step_container = new ScopedStepContainer( in StartParallelExecutors() local
510 [this, item, collector, cost_graph, step_container, in StartParallelExecutors()
514 delete step_container; in StartParallelExecutors()
522 args.step_container = step_container; in StartParallelExecutors()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor.h | 99 ScopedStepContainer* step_container = nullptr; member
|
| D | direct_session.cc | 638 args.step_container = &run_state.step_container; in RunInternal()
986 args.step_container = &run_state->step_container; in PRunSetup()
1756 : step_container(step_id, [devices, step_id](const string& name) { in RunState()
|
| D | direct_session.h | 205 ScopedStepContainer step_container; member
|
| D | function.cc | 526 opts.step_container = ctx->step_container(); in ComputeAsync()
1005 exec_args->step_container = run_opts.step_container; in ExecutorArgsFromOptions()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 675 ScopedStepContainer* step_container = nullptr; member
1215 ScopedStepContainer* step_container() const { in step_container() function
1216 return params_->step_container; in step_container()
|
| D | function.h | 693 ScopedStepContainer* step_container = nullptr; member
|