Searched refs:keep_dims_attr (Results 1 – 3 of 3) sorted by relevance
1043 auto keep_dims_attr = tf_all_op.keep_dimsAttr(); in matchAndRewrite() local1044 if (keep_dims_attr) keep_dims = keep_dims_attr.getValue(); in matchAndRewrite()1069 auto keep_dims_attr = tf_any_op.keep_dimsAttr(); in matchAndRewrite() local1070 if (keep_dims_attr) keep_dims = keep_dims_attr.getValue(); in matchAndRewrite()1095 auto keep_dims_attr = tf_max_op.keep_dimsAttr(); in matchAndRewrite() local1096 if (keep_dims_attr) keep_dims = keep_dims_attr.getValue(); in matchAndRewrite()1121 auto keep_dims_attr = tf_min_op.keep_dimsAttr(); in matchAndRewrite() local1122 if (keep_dims_attr) keep_dims = keep_dims_attr.getValue(); in matchAndRewrite()1147 auto keep_dims_attr = tf_mean_op.keep_dimsAttr(); in matchAndRewrite() local1148 if (keep_dims_attr) keep_dims = keep_dims_attr.getValue(); in matchAndRewrite()[all …]
1745 auto keep_dims_attr = tfl_any_op.keep_dimsAttr(); in matchAndRewrite() local1746 if (keep_dims_attr) keep_dims = keep_dims_attr.getValue(); in matchAndRewrite()1771 auto keep_dims_attr = tfl_max_op.keep_dimsAttr(); in matchAndRewrite() local1772 if (keep_dims_attr) keep_dims = keep_dims_attr.getValue(); in matchAndRewrite()1797 auto keep_dims_attr = tfl_min_op.keep_dimsAttr(); in matchAndRewrite() local1798 if (keep_dims_attr) keep_dims = keep_dims_attr.getValue(); in matchAndRewrite()1823 auto keep_dims_attr = tfl_prod_op.keep_dimsAttr(); in matchAndRewrite() local1824 if (keep_dims_attr) keep_dims = keep_dims_attr.getValue(); in matchAndRewrite()1849 auto keep_dims_attr = tfl_mean_op.keep_dimsAttr(); in matchAndRewrite() local1850 if (keep_dims_attr) keep_dims = keep_dims_attr.getValue(); in matchAndRewrite()[all …]
1466 const auto* keep_dims_attr = node.GetAttr(kAttrKeepDims); in KeepDims() local1467 if (keep_dims_attr != nullptr) { in KeepDims()1468 return keep_dims_attr->b(); in KeepDims()