Searched defs:axisAttr (Results 1 – 2 of 2) sorted by relevance
2182 IntegerAttr axisAttr = tf_one_hot_op.axisAttr(); in matchAndRewrite() local
3292 IntegerAttr axisAttr = tfl_one_hot_op.axisAttr(); in matchAndRewrite() local