Searched refs:is_single_element_op (Results 1 – 2 of 2) sorted by relevance
160 TensorShapeProto* output_tensor_shape, bool* is_single_element_op) const;
2466 bool* is_single_element_op) const { in IsReductionCandidateForSimplification()2505 *is_single_element_op = input_num_elements == 1 && output_num_elements == 1; in IsReductionCandidateForSimplification()2544 bool is_single_element_op = false; in SimplifyReduction() local2548 &is_single_element_op)) { in SimplifyReduction()2571 is_single_element_op && !keep_dims && (node->attr().count("T") > 0); in SimplifyReduction()