device_ordinal=[0])

static_shape = tf.placeholder(tf.int32, shape=([3]), name='static_size')

w = tf.Variable(tf.constant([[1.0], [2.0], [3.0]]), name='w')

w_dyn = tf.SetDynamicDimensionBounds(w, static_size])
tpu_call([w_dyn])
}];
  let arguments = (ins
    TF_Tensor:$input,
    TF_I32OrI64Tensor:$static_shape
  );

  let hasVerifier = 1;

  let results = (outs