Searched defs:reduction_ops (Results 1 – 5 of 5) sorted by relevance
71 llvm::SmallVectorImpl<mlir::TF::DTensorAllReduceOp>* reduction_ops, in CheckReduceAndSumOptimizationCriteria()125 llvm::SmallVector<mlir::TF::DTensorAllReduceOp, 4> reduction_ops; in OptimizeAllReduceAndSum() local
46 const unsigned int reduction_ops = reduction_axis.num_dimensions(); in validate_config() local
50 const unsigned int reduction_ops = reduction_axis.num_dimensions(); in validate_config() local
53 const int reduction_ops = reduction_axis.num_dimensions(); in calculate_reduce_mean_shape() local
1850 helper_expr_const_range reduction_ops() const { in reduction_ops() function1854 helper_expr_range reduction_ops() { in reduction_ops() function