bool ShapeInference::InferShapeForXlaGatherOp(XlaGatherOp op) {
  xla::Shape input_shape = xla::TypeToShape(op.getOperand().getType());
  if (input_shape == xla::Shape() || input_shape.is_unbounded_dynamic())
    return false;

  xla::Shape start_indices_shape =
      xla::TypeToShape(op.getStartIndices().getType());
  if (start_indices_shape == xla::Shape()) return false;

  xla::GatherDimensionNumbers gather_dim_numbers;

}

def TPUValidateInputsPass : Pass<"tf-tpu-validate-inputs", "ModuleOp"> {
  let summary = "Validates inputs to the TPU TF/XLA bridge";

  let description = [{
    This pass checks that the IR has valid input to TPU TF/XLA bridge.
    It checks the relations of multiple ops. Properties of single ops are
    checked by the 'verify' method of ops.
  }];

  let summary = "DynamicUpdateSlice.";
  let description = [{
    DynamicUpdateSlice op that have the same semantics with XLA
    DynamicUpdateSlice.
    Generates a result which is the value of the input array
    operand, with a slice update overwritten at start_indices.

    See https://www.tensorflow.org/xla/operation_semantics#dynamicupdateslice.
  }];

  let arguments = (ins