| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 249 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in Run()
338 ScopedStepContainer* step_container, std::vector<EagerKernelRet>* outputs, in PrepareForRun()
395 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in Run()
435 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in RunAsync()
|
| D | kernel_and_device.h | 219 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in RunAsync()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler.h | 60 ScopedStepContainer* step_container) in GraphCompiler()
|
| D | xla_compiler.cc | 134 auto step_container = absl::make_unique<ScopedStepContainer>( in ExecuteGraph() local
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_compat_request_state.h | 92 ScopedStepContainer* step_container() const { return step_container_.get(); } in step_container() function
|
| D | kernel_fallback_compat_request_state.cc | 68 tfrt::OwnedOrUnownedPtr<ScopedStepContainer> step_container, in KernelFallbackCompatRequestState()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 145 auto* step_container = ctx->step_container(); in GetStack() local
188 auto* step_container = ctx->step_container(); in Compute() local
|
| D | partitioned_function_ops.cc | 239 ScopedStepContainer* step_container = new ScopedStepContainer( in RunFunction() local
|
| /external/tensorflow/tensorflow/core/data/ |
| D | captured_function.cc | 785 ScopedStepContainer step_container( in Run() local
849 ScopedStepContainer step_container( in RunWithBorrowedArgs() local
905 ScopedStepContainer step_container( in RunInstantiated() local
952 ScopedStepContainer* step_container = new ScopedStepContainer( in RunAsync() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor.h | 99 ScopedStepContainer* step_container = nullptr; member
|
| D | direct_session.h | 205 ScopedStepContainer step_container; member
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 518 ScopedStepContainer* step_container = new ScopedStepContainer( in StartParallelExecutors() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 604 ScopedStepContainer* step_container = nullptr; member
1140 ScopedStepContainer* step_container() const { in step_container() function
|
| D | function.h | 815 ScopedStepContainer* step_container = nullptr; member
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 828 ScopedStepContainer step_container(opts.step_id, [ctx](const string& name) { in TryInit() local
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | c_api_tfrt.cc | 971 auto* step_container = GetEagerContext()->StepContainer(); in BuildFunctionRequestContext() local
|