Searched refs:cond_rank (Results 1 – 5 of 5) sorted by relevance
40 const int cond_rank = cond_shape.DimensionsCount(); in ResizeOutputTensor() local51 output_dims->data[1] = cond_rank; in ResizeOutputTensor()
822 const int32 cond_rank = c->Rank(cond); in __anon725c8f640402() local825 if (cond_rank == 0) { in __anon725c8f640402()832 if (cond_rank != 1) { in __anon725c8f640402()847 if (cond_rank == 1) { in __anon725c8f640402()
1582 const size_t cond_rank = input_condition_shape.DimensionsCount(); in SelectTrueCoords() local1584 std::vector<int> dims_to_count(cond_rank, 0); in SelectTrueCoords()1586 for (int i = 0; i < cond_rank; ++i) { in SelectTrueCoords()1596 for (int j = 0; j < cond_rank; ++j) { in SelectTrueCoords()1598 output_data[output_index * cond_rank + j] = coord_j; in SelectTrueCoords()
783 auto cond_rank = cond_tensor.getRank(); in Verify() local785 if (cond_rank == 0 || cond_rank == data_rank) return success(); in Verify()787 if (cond_rank == 1) { in Verify()
3037 cond_rank = array_ops.rank(cond)3039 cond_rank > 1, lambda: _inputs_with_flattening(pfor_input, [0, 1, 2]),3045 out = control_flow_ops.cond(cond_rank > 1,