return rewriter->create<arith::ConstantOp>(loc, values_type, constant_attr);
}

// Function to create a tf.SumOp to sum the element in 'value' reduced along the
// 'redux_axes'.
TF::SumOp createSumOp(Value value, Location loc,
                      llvm::ArrayRef<int32_t> redux_axes,
                      PatternRewriter* rewriter) {
  Value redux_op = createI32ConstantOp(redux_axes, loc, rewriter);