| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 132 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
147 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
209 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
216 ScopedStepContainer* step_container, const EagerKernelArgs& inputs, in RunAsync()
298 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
305 ScopedStepContainer* step_container, const EagerKernelArgs& inputs,
|
| D | kernel_and_device.cc | 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()
|
| D | context.h | 329 ScopedStepContainer* StepContainer();
640 std::unique_ptr<ScopedStepContainer> step_container_
|
| D | context.cc | 92 step_container_(std::make_unique<ScopedStepContainer>( in EagerContext()
390 step_container_.reset(new ScopedStepContainer( in ClearCachesAndDefaultExecutor()
612 step_container_.reset(new ScopedStepContainer( in EndStep()
617 ScopedStepContainer* EagerContext::StepContainer() { in StepContainer()
|
| D | execute.cc | 1126 ScopedStepContainer* container = ctx->StepContainer(); in EagerKernelExecute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler.h | 60 ScopedStepContainer* step_container) in GraphCompiler()
84 ScopedStepContainer* step_container_;
|
| D | xla_compiler.cc | 134 auto step_container = absl::make_unique<ScopedStepContainer>( in ExecuteGraph()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_mgr.h | 89 class ScopedStepContainer {
95 ScopedStepContainer(const int64 step_id, in ScopedStepContainer() function
102 ScopedStepContainer(const int64 step_id, in ScopedStepContainer() function
110 ~ScopedStepContainer() { CleanUp(); } in ~ScopedStepContainer()
893 ResourceHandle ScopedStepContainer::MakeResourceHandle(
902 Status ScopedStepContainer::Lookup(ResourceMgr* rm, const std::string& name,
908 Status ScopedStepContainer::LookupOrCreate(ResourceMgr* rm,
918 Status ScopedStepContainer::Create(ResourceMgr* rm, const std::string& name,
926 Status ScopedStepContainer::Delete(ResourceMgr* rm, const std::string& name) {
|
| D | function.h | 60 class ScopedStepContainer; variable
792 ScopedStepContainer* step_container = nullptr;
|
| D | op_kernel.h | 78 class ScopedStepContainer; variable
644 ScopedStepContainer* step_container = nullptr;
1195 ScopedStepContainer* step_container() const { in step_container()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil_test.cc | 29 TEST_F(OpsTestBase, ScopedStepContainer) { in TEST_F() argument
|
| D | partitioned_function_ops.cc | 239 ScopedStepContainer* step_container = new ScopedStepContainer( in RunFunction()
|
| D | ops_testutil.h | 167 std::unique_ptr<ScopedStepContainer> step_container_;
|
| D | ops_testutil.cc | 171 step_container_.reset(new ScopedStepContainer(0, [](const string&) {})); in RunOpKernel()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor.h | 105 ScopedStepContainer* step_container = nullptr;
|
| D | direct_session.h | 205 ScopedStepContainer step_container;
|
| D | executor.cc | 355 ScopedStepContainer* step_container_;
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 842 ScopedStepContainer step_container( in Run()
905 ScopedStepContainer step_container( in RunWithBorrowedArgs()
960 ScopedStepContainer step_container( in RunInstantiated()
1006 ScopedStepContainer* step_container = new ScopedStepContainer( in RunAsync()
|
| D | dataset_test_base.h | 759 std::unique_ptr<ScopedStepContainer> step_container_;
|
| D | iterator_ops.cc | 890 ScopedStepContainer step_container(opts.step_id, [ctx](const string& name) { in TryInit()
|
| D | dataset_test_base.cc | 507 absl::make_unique<ScopedStepContainer>(0, [](const string&) {}); in CreateOpKernelContext()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 518 ScopedStepContainer* step_container = new ScopedStepContainer( in StartParallelExecutors()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_with_tf2xla.cc | 348 std::unique_ptr<tensorflow::ScopedStepContainer> step_container_;
376 step_container_ = absl::make_unique<tensorflow::ScopedStepContainer>( in PrepareParams()
|