Home
last modified time | relevance | path

Searched refs:can_use_shapes (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer.h113 Status SimplifyArithmeticOps(bool can_use_shapes);
H A Darithmetic_optimizer.cc4289 Status ArithmeticOptimizer::SimplifyArithmeticOps(bool can_use_shapes) { in SimplifyArithmeticOps() argument
4307 if (options_.combine_add_to_addn && can_use_shapes) in SimplifyArithmeticOps()
4316 can_use_shapes) in SimplifyArithmeticOps()
4318 if (options_.minimize_broadcasts && can_use_shapes) in SimplifyArithmeticOps()
4320 if (options_.remove_identity_transpose && can_use_shapes) in SimplifyArithmeticOps()
4330 if (options_.replace_mul_with_tile && can_use_shapes) in SimplifyArithmeticOps()
4334 if (options_.remove_redundant_reshape && can_use_shapes) in SimplifyArithmeticOps()
4450 const bool can_use_shapes = status.ok(); in Optimize() local
4451 if (!can_use_shapes) { in Optimize()
4456 TF_RETURN_IF_ERROR(SimplifyArithmeticOps(can_use_shapes)); in Optimize()