TensorShape(), DT_FLOAT, false,
                      tensorflow::XlaLayoutPreference::kNoPreference));
  xla::ShapeProto shape_proto;
  shape_proto.set_element_type(xla::PrimitiveType::F32);
  shape_proto.mutable_layout();
  EXPECT_EQ(shape, xla::Shape(shape_proto));
  EXPECT_EQ(options.shape_determination_fns.layout_preference_fn(

  let hasCanonicalizer = 1;
}

def TFR_GetElementTypeOp : TFR_Op<"get_element_type", [Pure]> {
  let description = [{
    The `get_element_type` operation gets the element type of a tfr.tensor and
    returns !tfr.attr.

    Example:

    ```mlir
    %1 = "tfr.get_element_type"(%0) : !tfr.tensor -> !tfr.attr
    %1 = tfr.get_element_type %0 -> !tfr.attr
    ```
  }];

  %2 = shape.to_extent_tensor %1 : !shape.shape -> tensor<2xindex>
  func.return %2 : tensor<2xindex>
}

// -----

func.func @get_element_type(%arg0: !tfr.tensor) -> (!tfr.attr, !tfr.attr) {
  %0 = tfr.get_element_type %arg0 -> !tfr.attr
  %1 = "tfr.get_element_type"(%arg0) : (!tfr.tensor) -> !tfr.attr
  func.return %0, %1 : !tfr.attr, !tfr.attr
}

// -----

// CHECK-LABEL: from_tf_tensor