Searched refs:can_use_shapes (Results 1 – 2 of 2) sorted by relevance
112 Status SimplifyArithmeticOps(bool can_use_shapes);
3664 Status ArithmeticOptimizer::SimplifyArithmeticOps(bool can_use_shapes) { in SimplifyArithmeticOps() argument3681 if (options_.combine_add_to_addn && can_use_shapes) in SimplifyArithmeticOps()3689 if (options_.hoist_common_factor_out_of_aggregation && can_use_shapes) in SimplifyArithmeticOps()3691 if (options_.minimize_broadcasts && can_use_shapes) in SimplifyArithmeticOps()3693 if (options_.remove_identity_transpose && can_use_shapes) in SimplifyArithmeticOps()3817 const bool can_use_shapes = status.ok(); in Optimize() local3818 if (!can_use_shapes) { in Optimize()3823 TF_RETURN_IF_ERROR(SimplifyArithmeticOps(can_use_shapes)); in Optimize()