Searched refs:step_container_ (Results 1 – 11 of 11) sorted by relevance
64 step_container_(step_container) {} in GraphCompiler()84 ScopedStepContainer* step_container_; variable
336 params->step_container = step_container_; in PartiallySetupParams()
36 EXPECT_TRUE(step_container_ != nullptr); in TEST_F()
171 step_container_.reset(new ScopedStepContainer(0, [](const string&) {})); in RunOpKernel()172 params_->step_container = step_container_.get(); in RunOpKernel()
167 std::unique_ptr<ScopedStepContainer> step_container_; variable
348 std::unique_ptr<tensorflow::ScopedStepContainer> step_container_; member in mlir::mhlo::__anon6f7e6d590111::Tf2XlaRewriter376 step_container_ = absl::make_unique<tensorflow::ScopedStepContainer>( in PrepareParams()378 tensorflow::Status status = step_container_->Create( in PrepareParams()385 params_.step_container = step_container_.get(); in PrepareParams()
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()619 return step_container_.get(); in StepContainer()
640 std::unique_ptr<ScopedStepContainer> step_container_
355 ScopedStepContainer* step_container_; member in tensorflow::__anon6f8fc96b0111::ExecutorState403 step_container_(args.step_container), in ExecutorState()708 params.step_container = step_container_; in Process()
506 step_container_ = in CreateOpKernelContext()508 params->step_container = step_container_.get(); in CreateOpKernelContext()
759 std::unique_ptr<ScopedStepContainer> step_container_; variable