Searched refs:cond_rank (Results 1 – 4 of 4) sorted by relevance
38 const int cond_rank = cond_shape.DimensionsCount(); in ResizeOutputTensor() local49 output_dims->data[1] = cond_rank; in ResizeOutputTensor()
785 const int32 cond_rank = c->Rank(cond); in __anon57f5a5dd0502() local788 if (cond_rank == 0) { in __anon57f5a5dd0502()795 if (cond_rank != 1) { in __anon57f5a5dd0502()810 if (cond_rank == 1) { in __anon57f5a5dd0502()
2266 cond_rank = array_ops.rank(cond)2268 cond_rank > 1, lambda: _inputs_with_flattening(pfor_input, [0, 1, 2]),2274 out = control_flow_ops.cond(cond_rank > 1,
4338 const size_t cond_rank = input_condition_shape.DimensionsCount(); in SelectTrueCoords() local4340 std::vector<int> dims_to_count(cond_rank, 0); in SelectTrueCoords()4342 for (int i = 0; i < cond_rank; ++i) { in SelectTrueCoords()4352 for (int j = 0; j < cond_rank; ++j) { in SelectTrueCoords()4354 output_data[output_index * cond_rank + j] = coord_j; in SelectTrueCoords()