Searched refs:cond_vec (Results 1 – 4 of 4) sorted by relevance
82 TTypes<bool>::ConstVec cond_vec, in operator ()()85 const int batch = cond_vec.size(); in operator ()()102 cond_vec.reshape(reshape_dims) in operator ()()
355 TTypes<bool>::ConstVec cond_vec, in operator ()()358 const Eigen::DenseIndex batch = cond_vec.size(); in operator ()()367 cond_vec.reshape(reshape_dims) in operator ()()378 TTypes<bool>::ConstVec cond_vec, in operator ()()381 const size_t batch = cond_vec.size(); in operator ()()384 const bool* c = cond_vec.data(); in operator ()()
1374 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()