Searched refs:cond_vec (Results 1 – 4 of 4) sorted by relevance
61 TTypes<bool>::ConstVec cond_vec, in operator ()()64 const int batch = cond_vec.size(); in operator ()()86 cond_vec.reshape(reshape_dims) in operator ()()
262 TTypes<bool>::ConstVec cond_vec, in operator ()()265 const Eigen::DenseIndex batch = cond_vec.size(); in operator ()()279 cond_vec.reshape(reshape_dims) in operator ()()290 TTypes<bool>::ConstVec cond_vec, in operator ()()293 const size_t batch = cond_vec.size(); in operator ()()296 const bool* c = cond_vec.data(); in operator ()()
1204 TTypes<bool>::ConstVec cond_vec,
193 std::vector<HloInstruction*> cond_vec = {cond_param, slice, reshape, zero, in TEST_F() local199 schedule.set_sequence(cond_computation, cond_vec); in TEST_F()