Searched defs:keep_dims_attr (Results 1 – 3 of 3) sorted by relevance
1043 auto keep_dims_attr = tf_all_op.keep_dimsAttr(); in matchAndRewrite() local1069 auto keep_dims_attr = tf_any_op.keep_dimsAttr(); in matchAndRewrite() local1095 auto keep_dims_attr = tf_max_op.keep_dimsAttr(); in matchAndRewrite() local1121 auto keep_dims_attr = tf_min_op.keep_dimsAttr(); in matchAndRewrite() local1147 auto keep_dims_attr = tf_mean_op.keep_dimsAttr(); in matchAndRewrite() local1173 auto keep_dims_attr = tf_prod_op.keep_dimsAttr(); in matchAndRewrite() local1199 auto keep_dims_attr = tf_sum_op.keep_dimsAttr(); in matchAndRewrite() local
1745 auto keep_dims_attr = tfl_any_op.keep_dimsAttr(); in matchAndRewrite() local1771 auto keep_dims_attr = tfl_max_op.keep_dimsAttr(); in matchAndRewrite() local1797 auto keep_dims_attr = tfl_min_op.keep_dimsAttr(); in matchAndRewrite() local1823 auto keep_dims_attr = tfl_prod_op.keep_dimsAttr(); in matchAndRewrite() local1849 auto keep_dims_attr = tfl_mean_op.keep_dimsAttr(); in matchAndRewrite() local1875 auto keep_dims_attr = tfl_sum_op.keep_dimsAttr(); in matchAndRewrite() local
1466 const auto* keep_dims_attr = node.GetAttr(kAttrKeepDims); in KeepDims() local