| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | kernel_and_device.cc | 264 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in Run()
353 ScopedStepContainer* step_container, std::vector<EagerKernelRet>* outputs, in PrepareForRun()
413 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in Run()
458 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in RunAsync()
|
| H A D | kernel_and_device.h | 223 void RunAsync(ScopedStepContainer* step_container, in RunAsync()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | captured_function.cc | 794 ScopedStepContainer step_container( in Run() local
857 ScopedStepContainer step_container( in RunWithBorrowedArgs() local
912 ScopedStepContainer step_container( in RunInstantiated() local
959 ScopedStepContainer* step_container = new ScopedStepContainer( in RunAsync() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | stack.cc | 145 auto* step_container = ctx->step_container(); in GetStack() local
188 auto* step_container = ctx->step_container(); in Compute() local
|
| H A D | partitioned_function_ops.cc | 240 ScopedStepContainer* step_container = new ScopedStepContainer( in RunFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | graph_compiler.h | 60 ScopedStepContainer* step_container) in GraphCompiler()
|
| H A D | xla_compiler.cc | 136 auto step_container = std::make_unique<ScopedStepContainer>( in ExecuteGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | kernel_fallback_compat_request_state.cc | 64 tfrt::OwnedOrUnownedPtr<ScopedStepContainer> step_container, in KernelFallbackCompatRequestState()
|
| H A D | kernel_fallback_compat_request_state.h | 104 ScopedStepContainer* step_container() const { return step_container_.get(); } in step_container() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | executor.h | 102 ScopedStepContainer* step_container = nullptr; member
|
| H A D | direct_session.h | 205 ScopedStepContainer step_container; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | op_kernel.h | 617 ScopedStepContainer* step_container = nullptr; member
1151 ScopedStepContainer* step_container() const { in step_container() function
|
| H A D | function.h | 848 ScopedStepContainer* step_container = nullptr; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | graph_mgr.cc | 524 ScopedStepContainer* step_container = new ScopedStepContainer( in StartParallelExecutors() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | iterator_ops.cc | 821 ScopedStepContainer step_container(opts.step_id, [ctx](const string& name) { in TryInit() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/ |
| H A D | c_api_tfrt.cc | 1012 auto* step_container = GetEagerContext()->StepContainer(); in BuildFunctionRequestContext() local
|