Home
last modified time | relevance | path

Searched defs:step_container (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc155 ScopedStepContainer step_container(0, [this](const string& name) { in Run() local
167 ScopedStepContainer step_container(0, [&devices](const string& name) { in Run() local
206 Status KernelAndDeviceOp::Run(ScopedStepContainer* step_container, in Run()
305 ScopedStepContainer* step_container, in Run()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.h60 ScopedStepContainer* step_container) in GraphCompiler()
Dxla_compiler.cc127 auto step_container = absl::make_unique<ScopedStepContainer>( in ExecuteGraph() local
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc305 ScopedStepContainer step_container( in Run() local
341 ScopedStepContainer step_container( in RunWithBorrowedArgs() local
377 ScopedStepContainer step_container( in RunInstantiated() local
421 ScopedStepContainer* step_container = new ScopedStepContainer( in RunAsync() local
Diterator_ops.cc879 ScopedStepContainer step_container(opts.step_id, [ctx](const string& name) { in TryInit() local
/external/tensorflow/tensorflow/core/kernels/
Dstack.cc144 auto* step_container = ctx->step_container(); in GetStack() local
188 auto* step_container = ctx->step_container(); in Compute() local
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.h94 ScopedStepContainer* step_container = nullptr; member
Ddirect_session.h190 ScopedStepContainer step_container; member
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc477 ScopedStepContainer* step_container = new ScopedStepContainer( in StartParallelExecutors() local
/external/tensorflow/tensorflow/core/framework/
Dfunction.h620 ScopedStepContainer* step_container = nullptr; member
Dop_kernel.h627 ScopedStepContainer* step_container = nullptr; member
1163 ScopedStepContainer* step_container() const { in step_container() function