Home
last modified time | relevance | path

Searched refs:step_container_ (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.h64 step_container_(step_container) {} in GraphCompiler()
84 ScopedStepContainer* step_container_; variable
Dgraph_compiler.cc309 params->step_container = step_container_; in PartiallySetupParams()
/external/tensorflow/tensorflow/core/kernels/
Dops_testutil.h172 step_container_.reset(new ScopedStepContainer(0, [](const string&) {})); in RunOpKernel()
173 params_->step_container = step_container_.get(); in RunOpKernel()
230 std::unique_ptr<ScopedStepContainer> step_container_; variable
Dops_testutil_test.cc34 EXPECT_TRUE(step_container_ != nullptr); in TEST_F()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.cc281 if (step_container_ == nullptr) { in StartStep()
282 step_container_.reset( in StartStep()
295 step_container_.reset(); in EndStep()
304 return step_container_.get(); in StepContainer()
Dcontext.h300 std::unique_ptr<ScopedStepContainer> step_container_ GUARDED_BY(metadata_mu_);
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.cc189 step_container_ = in CreateOpKernelContext()
191 params_->step_container = step_container_.get(); in CreateOpKernelContext()
Ddataset_test_base.h185 std::unique_ptr<ScopedStepContainer> step_container_; variable
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.cc1252 ScopedStepContainer* step_container_; member in tensorflow::__anon675242740111::ExecutorState
1387 step_container_(args.step_container), in ExecutorState()
1637 params.step_container = step_container_; in Process()