Searched refs:mlir_type (Results 1 – 3 of 3) sorted by relevance
188 StatusOr<mlir::Type> mlir_type = in TEST() local190 ASSERT_TRUE(mlir_type.ok()); in TEST()191 mlir::Type type = mlir_type.ConsumeValueOrDie(); in TEST()205 StatusOr<mlir::Type> mlir_type = in TEST() local207 ASSERT_TRUE(mlir_type.ok()); in TEST()208 mlir::Type type = mlir_type.ConsumeValueOrDie(); in TEST()
307 Type mlir_type; in SetAttrType() local309 TF_RETURN_IF_ERROR(ConvertDataType(dtype, builder, &mlir_type)); in SetAttrType()310 attrs_[attr_name] = TypeAttr::get(mlir_type); in SetAttrType()
58 mlir_type = _get_mlir_type(ctx.attr.type)71 mlir_type,