supported_ops->insert(OperationName(
      mlir::TF::XlaSelectAndScatterOp::getOperationName(), context));
  supported_ops->insert(
      OperationName(mlir::TF::SymbolicGradientOp::getOperationName(), context));
  supported_ops->insert(OperationName(
      mlir::TF::XlaVariadicReduceOp::getOperationName(), context));
  supported_ops->insert(OperationName(

      for (const int dy_index : no_grad_dy_indices) {
        dy[dy_index] = ops::ZerosLike(scope_, Output(n, dy_index));
      }
    }

    // TODO(andydavis) Add option to encapsulate grad function in
    // SymbolicGradientOp (as opposed to inlining into the graph).
    std::vector<Output> dx;
    TF_RETURN_IF_ERROR(CallGradFunction(Operation(n), dy, &dx));

    // Backprop along the in edges.