Searched refs:batch_attr (Results 1 – 1 of 1) sorted by relevance
3174 if (auto batch_attr = tfl_gather_op.batch_dimsAttr()) { in matchAndRewrite() local3175 batch_dims = static_cast<int32_t>(batch_attr.getInt()); in matchAndRewrite()