Searched refs:TypeToPrimitiveType (Results 1 – 5 of 5) sorted by relevance
45 PrimitiveType TypeToPrimitiveType(mlir::Type type) { in TypeToPrimitiveType() function101 PrimitiveType ptype = TypeToPrimitiveType(type); in TypeToShape()113 PrimitiveType primitive_type = TypeToPrimitiveType(element_type); in TypeToShape()125 PrimitiveType primitive_type = TypeToPrimitiveType(element_type); in TypeToShape()172 PrimitiveType primitive_type = TypeToPrimitiveType(element_type); in TypeToShape()
69 EXPECT_EQ(TypeToPrimitiveType(b.getF32Type()), PrimitiveType::F32); in TEST()70 EXPECT_EQ(TypeToPrimitiveType(b.getIntegerType(1)), PrimitiveType::PRED); in TEST()71 EXPECT_EQ(TypeToPrimitiveType(b.getIntegerType(17)), in TEST()
44 PrimitiveType TypeToPrimitiveType(mlir::Type type);
600 operand, xla::TypeToPrimitiveType(getElementTypeOrSelf(op.getType()))); in ExportXlaOp()738 operand, xla::TypeToPrimitiveType(getElementTypeOrSelf(op.getType()))); in ExportXlaOp()
238 auto xla_element_type = ::xla::TypeToPrimitiveType(element_type); in SetFrontendAttributes()