Home
last modified time | relevance | path

Searched refs:step_container (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dstack.cc145 auto* step_container = ctx->step_container(); in GetStack() local
146 if (step_container == nullptr) { in GetStack()
149 TF_RETURN_IF_ERROR(step_container->Lookup(rm, key, stack)); in GetStack()
188 auto* step_container = ctx->step_container(); in Compute() local
189 OP_REQUIRES(ctx, step_container != nullptr, in Compute()
192 OP_REQUIRES_OK(ctx, step_container->Create(rm, key, stack)); in Compute()
208 ctx->step_container()->MakeResourceHandle<Stack>(key, *ctx->device()); in Compute()
Dpartitioned_function_ops.cc239 ScopedStepContainer* step_container = new ScopedStepContainer( in RunFunction() local
243 run_opts.step_container = step_container; in RunFunction()
263 step_container](const Status& status) { in RunFunction()
275 delete step_container; in RunFunction()
Dvariable_ops.cc119 context->step_container()->Create(rm, unique_name, tmp_var)); in Compute()
165 context, context->step_container()->Delete<TemporaryVariableOp::TmpVar>( in Compute()
Dfunction_ops.cc230 opts.step_container = ctx->step_container(); in ComputeAsync()
Dops_testutil.cc172 params_->step_container = step_container_.get(); in RunOpKernel()
Dtensor_array_ops.cc84 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()
308 TF_RETURN_IF_ERROR(ctx->step_container()->Lookup( in CreateTensorArray()
364 Status s = ctx->step_container()->LookupOrCreate<TensorArray>( in CreateTensorArray()
Dbatch_kernels.cc159 opts.step_container = last_task_context->step_container(); in ProcessFuncBatchImpl()
Dfunctional_ops.cc123 opts->step_container = ctx->step_container(); in SetRunOptions()
Dtensor_array.h348 return ctx->step_container()->MakeResourceHandle<TensorArray>( in resource_handle()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h132 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
147 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
209 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
216 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in RunAsync() argument
222 done(Run(step_container, inputs, outputs, cancellation_manager, in RunAsync()
298 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
305 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
Dkernel_and_device.cc243 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in Run() argument
282 params.step_container = step_container; in Run()
322 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in Run() argument
329 RunAsync(step_container, inputs, outputs, cancellation_manager, in Run()
339 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in RunAsync() argument
384 opts->step_container = step_container; in RunAsync()
394 [opts, rendezvous, local_cm, step_container, this, in RunAsync()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.h60 ScopedStepContainer* step_container) in GraphCompiler() argument
64 step_container_(step_container) {} in GraphCompiler()
Dxla_context.cc47 TF_CHECK_OK(ctx->step_container()->Lookup(ctx->resource_manager(), in Get()
Dgraph_compiler.cc336 params->step_container = step_container_; in PartiallySetupParams()
Dxla_compiler.cc134 auto step_container = absl::make_unique<ScopedStepContainer>( in ExecuteGraph() local
138 TF_RETURN_IF_ERROR(step_container->Create(device->resource_manager(), in ExecuteGraph()
142 GraphCompiler graph_compiler(device, graph.get(), flib, step_container.get()); in ExecuteGraph()
145 step_container.reset(); in ExecuteGraph()
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc842 ScopedStepContainer step_container( in Run() local
846 f_opts.step_container = &step_container; in Run()
905 ScopedStepContainer step_container( in RunWithBorrowedArgs() local
909 f_opts.step_container = &step_container; in RunWithBorrowedArgs()
960 ScopedStepContainer step_container( in RunInstantiated() local
964 f_opts.step_container = &step_container; in RunInstantiated()
1006 ScopedStepContainer* step_container = new ScopedStepContainer( in RunAsync() local
1010 f_opts.step_container = step_container; in RunAsync()
1029 [this, rets, step_container, raw_cancellation_manager, frame, node, in RunAsync()
1036 delete step_container; in RunAsync()
Dsingle_threaded_executor.cc297 params.step_container = args.step_container; in Run()
Diterator_ops.cc890 ScopedStepContainer step_container(opts.step_id, [ctx](const string& name) { in TryInit() local
893 opts.step_container = &step_container; in TryInit()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc518 ScopedStepContainer* step_container = new ScopedStepContainer( in StartParallelExecutors() local
524 [this, item, collector, cost_graph, step_container, in StartParallelExecutors()
528 delete step_container; in StartParallelExecutors()
536 args.step_container = step_container; in StartParallelExecutors()
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.h105 ScopedStepContainer* step_container = nullptr; member
Ddirect_session.cc650 args.step_container = &run_state.step_container; in RunInternal()
1000 args.step_container = &run_state->step_container; in PRunSetup()
1768 : step_container(step_id, [devices, step_id](const string& name) { in RunState()
Dfunction.cc531 opts.step_container = ctx->step_container(); in ComputeAsync()
986 exec_args->step_container = run_opts.step_container; in ExecutorArgsFromOptions()
Ddirect_session.h205 ScopedStepContainer step_container; member
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h644 ScopedStepContainer* step_container = nullptr; member
1195 ScopedStepContainer* step_container() const { in step_container() function
1196 return params_->step_container; in step_container()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.cc674 if (ctx->step_container() == nullptr) { in ComputeInternal()
679 ctx->step_container()->LookupOrCreate<CompilationRefHolder>( in ComputeInternal()

12