Searched refs:XlaOpRegistrationBuilder (Results 1 – 2 of 2) sorted by relevance
173 friend class XlaOpRegistrationBuilder; variable276 class XlaOpRegistrationBuilder {279 static XlaOpRegistrationBuilder Name(absl::string_view name);282 XlaOpRegistrationBuilder& Device(absl::string_view devices);283 XlaOpRegistrationBuilder& Device(absl::Span<const absl::string_view> devices);287 XlaOpRegistrationBuilder& TypeConstraint(absl::string_view attr_name,290 XlaOpRegistrationBuilder& TypeConstraint(absl::string_view attr_name,295 XlaOpRegistrationBuilder& CompilationOnly();298 XlaOpRegistrationBuilder& AllowResourceTypes();301 XlaOpRegistrationBuilder& AllowVariantTypes();[all …]
460 XlaOpRegistrationBuilder::XlaOpRegistrationBuilder(absl::string_view name) { in XlaOpRegistrationBuilder() function in tensorflow::XlaOpRegistrationBuilder465 XlaOpRegistrationBuilder XlaOpRegistrationBuilder::Name( in Name()467 XlaOpRegistrationBuilder registration(name); in Name()471 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::Device( in Device()480 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::Device( in Device()487 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::CompilationOnly() { in CompilationOnly()492 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::AllowResourceTypes() { in AllowResourceTypes()497 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::AllowVariantTypes() { in AllowVariantTypes()502 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::TypeConstraint( in TypeConstraint()510 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::TypeConstraint( in TypeConstraint()[all …]