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()
39 TF_RETURN_IF_ERROR(body_thunk_sequence_->Initialize(executable, executor)); in Initialize()
52 TF_RETURN_IF_ERROR(body_thunk_sequence_->ExecuteOnStream(buffer_allocations, in ExecuteOnStream()
Dwhile_thunk.cc39 body_thunk_sequence_(absl::make_unique<SequentialThunk>( in WhileThunk()
46 TF_RETURN_IF_ERROR(body_thunk_sequence_->Initialize(executable, executor)); in Initialize()
86 TF_RETURN_IF_ERROR(body_thunk_sequence_->ExecuteOnStream(buffer_allocations, in ExecuteOnStream()
Dfor_thunk.h48 std::unique_ptr<SequentialThunk> body_thunk_sequence_; variable
Dwhile_thunk.h58 std::unique_ptr<SequentialThunk> body_thunk_sequence_; variable