Home
last modified time | relevance | path

Searched refs:step_container (Results 1 – 25 of 38) 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.cc240 ScopedStepContainer* step_container = new ScopedStepContainer( in RunFunction() local
244 run_opts.step_container = step_container; in RunFunction()
264 step_container](const Status& status) { in RunFunction()
277 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.cc244 opts.step_container = ctx->step_container(); in ComputeAsync()
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()
Dops_testutil.cc172 params_->step_container = step_container_.get(); in RunOpKernel()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc264 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in Run() argument
304 params.step_container = step_container; in Run()
353 ScopedStepContainer* step_container, std::vector<EagerKernelRet>* outputs, in PrepareForRun() argument
399 opts->step_container = step_container; in PrepareForRun()
413 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in Run() argument
425 RunAsync(step_container, inputs, outputs, cancellation_manager, in Run()
435 PrepareForRun(step_container, outputs, cancellation_manager, in Run()
458 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in RunAsync() argument
467 step_container, outputs, cancellation_manager, eager_func_params, in RunAsync()
Dkernel_and_device.h138 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
154 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
216 Status Run(ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
223 void RunAsync(ScopedStepContainer* step_container, in RunAsync() argument
231 done(Run(step_container, inputs, outputs, cancellation_manager, in RunAsync()
318 Status Run(ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
325 void RunAsync(ScopedStepContainer* step_container,
349 ScopedStepContainer* step_container, std::vector<EagerKernelRet>* outputs,
/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()
Dxla_compiler.cc136 auto step_container = std::make_unique<ScopedStepContainer>( in ExecuteGraph() local
140 TF_RETURN_IF_ERROR(step_container->Create(device->resource_manager(), in ExecuteGraph()
144 GraphCompiler graph_compiler(device, graph.get(), flib, step_container.get()); in ExecuteGraph()
147 step_container.reset(); in ExecuteGraph()
Dgraph_compiler.cc341 params->step_container = step_container_; in PartiallySetupParams()
/external/tensorflow/tensorflow/core/data/
Dcaptured_function.cc794 ScopedStepContainer step_container( in Run() local
798 f_opts.step_container = &step_container; in Run()
857 ScopedStepContainer step_container( in RunWithBorrowedArgs() local
861 f_opts.step_container = &step_container; in RunWithBorrowedArgs()
912 ScopedStepContainer step_container( in RunInstantiated() local
916 f_opts.step_container = &step_container; in RunInstantiated()
959 ScopedStepContainer* step_container = new ScopedStepContainer( in RunAsync() local
963 f_opts.step_container = step_container; in RunAsync()
982 [this, rets, step_container, raw_cancellation_manager, frame, node, in RunAsync()
989 delete step_container; in RunAsync()
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dkernel_fallback_utils.cc38 params.step_container = fallback_request_state.step_container(); in SetUpParams()
Dkernel_fallback_compat_request_state.cc64 tfrt::OwnedOrUnownedPtr<ScopedStepContainer> step_container, in KernelFallbackCompatRequestState() argument
72 step_container_(std::move(step_container)), in KernelFallbackCompatRequestState()
Dkernel_fallback_compat_request_state.h77 tfrt::OwnedOrUnownedPtr<ScopedStepContainer> step_container,
104 ScopedStepContainer* step_container() const { return step_container_.get(); } in step_container() function
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc524 ScopedStepContainer* step_container = new ScopedStepContainer( in StartParallelExecutors() local
530 [this, item, collector, cost_graph, step_container, in StartParallelExecutors()
534 delete step_container; in StartParallelExecutors()
542 args.step_container = step_container; in StartParallelExecutors()
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.h102 ScopedStepContainer* step_container = nullptr; member
Dsingle_threaded_executor.cc321 params.step_container = args.step_container; in Run()
Ddirect_session.cc666 args.step_container = &run_state.step_container; in RunInternal()
1017 args.step_container = &run_state->step_container; in PRunSetup()
1785 : step_container(step_id, [devices, step_id](const string& name) { in RunState()
Dfunction.cc541 opts.step_container = ctx->step_container(); in ComputeAsync()
1018 exec_args->step_container = run_opts.step_container; in ExecutorArgsFromOptions()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.cc271 if (ctx->step_container() == nullptr) { in ComputeInternal()
276 ctx->step_container()->LookupOrCreate<CompilationRefHolder>( in ComputeInternal()
Dtpu_functional_ops.cc1413 opts.step_container = ctx->step_container(); in InitializeVarOnTPU()
1583 opts.step_container = ctx->step_container(); in InitializeShardedVarOnTPU()
2683 opts.step_container = ctx->step_container(); in ExecuteFunctions()
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.cc820 ScopedStepContainer step_container(opts.step_id, [ctx](const string& name) { in TryInit() local
823 opts.step_container = &step_container; in TryInit()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h617 ScopedStepContainer* step_container = nullptr; member
1151 ScopedStepContainer* step_container() const { in step_container() function
1152 return params_->step_container; in step_container()

12