Searched refs:is_single_element_op (Results 1 – 2 of 2) sorted by relevance
233 TensorShapeProto* output_tensor_shape, bool* is_single_element_op) const;
2746 bool* is_single_element_op) const { in IsReductionCandidateForSimplification()2775 *is_single_element_op = input_num_elements == 1 && output_num_elements == 1; in IsReductionCandidateForSimplification()2839 bool is_single_element_op = false; in SimplifyReduction() local2843 &is_single_element_op)) { in SimplifyReduction()2866 is_single_element_op && !keep_dims && (node->attr().count("T") > 0); in SimplifyReduction()