Searched defs:step_container (Results 1 – 11 of 11) sorted by relevance
60 ScopedStepContainer* step_container) in GraphCompiler()
307 auto step_container = xla::MakeUnique<ScopedStepContainer>( in ExecuteGraph() local
198 ScopedStepContainer step_container(f_opts.step_id, [ctx](const string& name) { in Run() local232 ScopedStepContainer step_container(f_opts.step_id, [ctx](const string& name) { in RunWithBorrowedArgs() local278 auto step_container = new ScopedStepContainer( in RunAsync() local
776 ScopedStepContainer step_container(opts.step_id, [ctx](const string& name) { in TryInit() local
153 auto* step_container = ctx->step_container(); in GetStack() local198 auto* step_container = ctx->step_container(); in Compute() local
184 auto* step_container = in ExecuteFunction() local
91 ScopedStepContainer* step_container = nullptr; member
173 ScopedStepContainer step_container; member
480 ScopedStepContainer* step_container = nullptr; member
529 ScopedStepContainer* step_container = nullptr; member1026 ScopedStepContainer* step_container() const { in step_container() function
455 ScopedStepContainer* step_container = new ScopedStepContainer( in StartParallelExecutors() local