Searched refs:XlaOpRegistrationBuilder (Results 1 – 2 of 2) sorted by relevance
217 friend class XlaOpRegistrationBuilder; variable326 class XlaOpRegistrationBuilder {329 static XlaOpRegistrationBuilder Name(absl::string_view name);332 XlaOpRegistrationBuilder& Device(absl::string_view devices);333 XlaOpRegistrationBuilder& Device(absl::Span<const absl::string_view> devices);337 XlaOpRegistrationBuilder& TypeConstraint(absl::string_view attr_name,340 XlaOpRegistrationBuilder& TypeConstraint(absl::string_view attr_name,345 XlaOpRegistrationBuilder& CompilationOnly();348 XlaOpRegistrationBuilder& AllowResourceTypes();351 XlaOpRegistrationBuilder& AllowVariantTypes();[all …]
481 XlaOpRegistrationBuilder::XlaOpRegistrationBuilder(absl::string_view name) { in XlaOpRegistrationBuilder() function in tensorflow::XlaOpRegistrationBuilder486 XlaOpRegistrationBuilder XlaOpRegistrationBuilder::Name( in Name()488 XlaOpRegistrationBuilder registration(name); in Name()492 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::Device( in Device()501 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::Device( in Device()508 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::CompilationOnly() { in CompilationOnly()513 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::AllowResourceTypes() { in AllowResourceTypes()518 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::AllowVariantTypes() { in AllowVariantTypes()523 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::AllowStringType() { in AllowStringType()528 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::TypeConstraint( in TypeConstraint()[all …]