Searched refs:cond_rank (Results 1 – 5 of 5) sorted by relevance
42 const int cond_rank = cond_shape.DimensionsCount(); in ResizeOutputTensor() local53 output_dims->data[1] = cond_rank; in ResizeOutputTensor()
777 const size_t cond_rank = input_condition_shape.DimensionsCount(); in SelectTrueCoords() local779 std::vector<int> dims_to_count(cond_rank, 0); in SelectTrueCoords()781 for (int i = 0; i < cond_rank; ++i) { in SelectTrueCoords()791 for (int j = 0; j < cond_rank; ++j) { in SelectTrueCoords()793 output_data[output_index * cond_rank + j] = coord_j; in SelectTrueCoords()
855 const int32_t cond_rank = c->Rank(cond); in __anon328d24420402() local858 if (cond_rank == 0) { in __anon328d24420402()865 if (cond_rank != 1) { in __anon328d24420402()880 if (cond_rank == 1) { in __anon328d24420402()
950 auto cond_rank = cond_tensor.getRank(); in verify() local952 if (cond_rank == 0 || cond_rank == data_rank) return success(); in verify()954 if (cond_rank == 1) { in verify()
3255 cond_rank = array_ops.rank(cond)3257 cond_rank > 1, lambda: _inputs_with_flattening(pfor_input, [0, 1, 2]),3263 out = smart_cond.smart_cond(cond_rank > 1,