Home
last modified time | relevance | path

Searched refs:branch_thunks_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dconditional_thunk.cc39 branch_thunks_.reserve(branch_thunk_sequences.size()); in ConditionalThunk()
41 branch_thunks_.emplace_back( in ConditionalThunk()
49 TF_RET_CHECK(branch_thunks_.size() == 2); in Initialize()
51 TF_RET_CHECK(!branch_thunks_.empty()); in Initialize()
53 for (auto& branch_thunk : branch_thunks_) { in Initialize()
91 TF_RETURN_IF_ERROR(branch_thunks_[branch_index]->ExecuteOnStream( in ExecuteOnStream()
Dconditional_thunk.h64 std::vector<std::unique_ptr<SequentialThunk>> branch_thunks_; variable