Searched defs:reduction_ops (Results 1 – 7 of 7) sorted by relevance
215 def reduction_ops(self): member in ReductionOpsModule
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
24420 reduction_ops = [op for op in ops_and_refs if isinstance(op, ReductionOpInfo)] variable