Searched refs:step_container_ (Results 1 – 9 of 9) sorted by relevance
64 step_container_(step_container) {} in GraphCompiler()84 ScopedStepContainer* step_container_; variable
309 params->step_container = step_container_; in PartiallySetupParams()
172 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
34 EXPECT_TRUE(step_container_ != nullptr); in TEST_F()
281 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()
300 std::unique_ptr<ScopedStepContainer> step_container_ GUARDED_BY(metadata_mu_);
189 step_container_ = in CreateOpKernelContext()191 params_->step_container = step_container_.get(); in CreateOpKernelContext()
185 std::unique_ptr<ScopedStepContainer> step_container_; variable
1252 ScopedStepContainer* step_container_; member in tensorflow::__anon675242740111::ExecutorState1387 step_container_(args.step_container), in ExecutorState()1637 params.step_container = step_container_; in Process()