Home
last modified time | relevance | path

Searched refs:body_thunk_sequence_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfor_thunk.cc31 body_thunk_sequence_(absl::make_unique<SequentialThunk>( in ForThunk()
40 TF_RETURN_IF_ERROR(body_thunk_sequence_->Initialize(executable, executor)); in Initialize()
51 TF_RETURN_IF_ERROR(body_thunk_sequence_->ExecuteOnStream(params)); in ExecuteOnStream()
Dwhile_thunk.cc41 body_thunk_sequence_(absl::make_unique<SequentialThunk>( in WhileThunk()
50 TF_RETURN_IF_ERROR(body_thunk_sequence_->Initialize(executable, executor)); in Initialize()
91 TF_RETURN_IF_ERROR(body_thunk_sequence_->ExecuteOnStream(params)); in ExecuteOnStream()
Dfor_thunk.h46 std::unique_ptr<SequentialThunk> body_thunk_sequence_; variable
Dwhile_thunk.h58 std::unique_ptr<SequentialThunk> body_thunk_sequence_; variable