Searched defs:step_container (Results 1 – 13 of 13) sorted by relevance
60 ScopedStepContainer* step_container) in GraphCompiler()
134 auto step_container = absl::make_unique<ScopedStepContainer>( in ExecuteGraph() local
243 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in Run()322 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in Run()339 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in RunAsync()
216 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in RunAsync()
145 auto* step_container = ctx->step_container(); in GetStack() local188 auto* step_container = ctx->step_container(); in Compute() local
239 ScopedStepContainer* step_container = new ScopedStepContainer( in RunFunction() local
105 ScopedStepContainer* step_container = nullptr; member
205 ScopedStepContainer step_container; member
842 ScopedStepContainer step_container( in Run() local905 ScopedStepContainer step_container( in RunWithBorrowedArgs() local960 ScopedStepContainer step_container( in RunInstantiated() local1006 ScopedStepContainer* step_container = new ScopedStepContainer( in RunAsync() local
890 ScopedStepContainer step_container(opts.step_id, [ctx](const string& name) { in TryInit() local
518 ScopedStepContainer* step_container = new ScopedStepContainer( in StartParallelExecutors() local
644 ScopedStepContainer* step_container = nullptr; member1195 ScopedStepContainer* step_container() const { in step_container() function
792 ScopedStepContainer* step_container = nullptr; member