Searched refs:true_indices (Results 1 – 3 of 3) sorted by relevance
1808 std::set<int64>* true_indices) { in ReductionShapeHelper() argument1823 true_indices->insert(wrapped_index); in ReductionShapeHelper()1858 std::set<int64> true_indices; in ReductionShape() local1861 input_rank, &true_indices)); in ReductionShape()1864 input_rank, &true_indices)); in ReductionShape()1872 if (true_indices.count(i) > 0) { in ReductionShape()
574 std::set<int32> true_indices; in ProcessTensorFlowReductionOperator() local587 true_indices.insert(wrapped_index); in ProcessTensorFlowReductionOperator()593 if (true_indices.count(i) > 0) { in ProcessTensorFlowReductionOperator()
4244 false_indices, true_indices = data_flow_ops.dynamic_partition(4252 else_indices, then_indices = false_indices, true_indices4258 then_outputs = _outputs_for_branch(then_branch.name, true_indices,