Searched refs:ScopedStepContainer (Results 1 – 20 of 20) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler.h | 60 ScopedStepContainer* step_container) in GraphCompiler()
84 ScopedStepContainer* step_container_;
|
| D | xla_compiler.cc | 127 auto step_container = absl::make_unique<ScopedStepContainer>( in ExecuteGraph()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 86 virtual Status Run(ScopedStepContainer* step_container,
151 Status Run(ScopedStepContainer* step_container,
212 Status Run(ScopedStepContainer* step_container,
|
| D | kernel_and_device.cc | 155 ScopedStepContainer step_container(0, [this](const string& name) { in Run()
167 ScopedStepContainer step_container(0, [&devices](const string& name) { in Run()
206 Status KernelAndDeviceOp::Run(ScopedStepContainer* step_container, in Run()
305 ScopedStepContainer* step_container, in Run()
|
| D | context.h | 198 ScopedStepContainer* StepContainer();
300 std::unique_ptr<ScopedStepContainer> step_container_ GUARDED_BY(metadata_mu_);
|
| D | context.cc | 283 new ScopedStepContainer(0, [this](const string& name) { in StartStep()
299 ScopedStepContainer* EagerContext::StepContainer() { in StepContainer()
|
| D | execute.cc | 942 ScopedStepContainer* container = ctx->StepContainer(); in EagerKernelExecute()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 305 ScopedStepContainer step_container( in Run()
341 ScopedStepContainer step_container( in RunWithBorrowedArgs()
377 ScopedStepContainer step_container( in RunInstantiated()
421 ScopedStepContainer* step_container = new ScopedStepContainer( in RunAsync()
|
| D | dataset_test_base.h | 185 std::unique_ptr<ScopedStepContainer> step_container_;
|
| D | dataset_test_base.cc | 190 absl::make_unique<ScopedStepContainer>(0, [](const string&) {}); in CreateOpKernelContext()
|
| D | iterator_ops.cc | 879 ScopedStepContainer step_container(opts.step_id, [ctx](const string& name) { in TryInit()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil_test.cc | 27 TEST_F(OpsTestBase, ScopedStepContainer) { in TEST_F() argument
|
| D | ops_testutil.h | 172 step_container_.reset(new ScopedStepContainer(0, [](const string&) {})); in RunOpKernel()
230 std::unique_ptr<ScopedStepContainer> step_container_;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_mgr.h | 87 class ScopedStepContainer {
93 ScopedStepContainer(const int64 step_id, in ScopedStepContainer() function
97 ScopedStepContainer(const int64 step_id, in ScopedStepContainer() function
103 ~ScopedStepContainer() { cleanup_(name_); } in ~ScopedStepContainer()
|
| D | function.h | 47 class ScopedStepContainer; variable
620 ScopedStepContainer* step_container = nullptr;
|
| D | op_kernel.h | 74 class ScopedStepContainer; variable
627 ScopedStepContainer* step_container = nullptr;
1163 ScopedStepContainer* step_container() const { in step_container()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor.h | 94 ScopedStepContainer* step_container = nullptr;
|
| D | direct_session.h | 190 ScopedStepContainer step_container;
|
| D | executor.cc | 1252 ScopedStepContainer* step_container_;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 477 ScopedStepContainer* step_container = new ScopedStepContainer( in StartParallelExecutors()
|