Searched refs:registration_ (Results 1 – 5 of 5) sorted by relevance
461 registration_.reset(new XlaOpRegistry::OpRegistration); in XlaOpRegistrationBuilder()462 registration_->name = string(name); in XlaOpRegistrationBuilder()473 registration_->has_device_whitelist = true; in Device()475 registration_->device_whitelist.emplace(device); in Device()482 registration_->has_device_whitelist = true; in Device()483 registration_->device_whitelist.emplace(device); in Device()488 registration_->compilation_only = true; in CompilationOnly()493 registration_->allow_resource_types = true; in AllowResourceTypes()498 registration_->allow_variant_types = true; in AllowVariantTypes()505 registration_->type_constraints[string(attr_name)]; in TypeConstraint()[all …]
317 std::unique_ptr<XlaOpRegistry::OpRegistration> registration_; variable
263 : registration_(std::make_unique<Registration>(this)), in SequenceLocalSyncEventWatcher()269 registration_->shared_state()->SignalForWatcher(this); in SignalEvent()273 registration_->shared_state()->ResetForWatcher(this); in ResetEvent()282 return registration_->shared_state()->SyncWatch( in SyncWatch()283 this, registration_->watcher_state(), should_stop); in SyncWatch()
121 : op_(op), registration_(*registration) { in SingleOpResolver()122 registration_.builtin_code = static_cast<int32_t>(op); in SingleOpResolver()123 registration_.version = 1; in SingleOpResolver()128 return ®istration_; in FindOp()138 TfLiteRegistration registration_; variable
60 const std::unique_ptr<Registration> registration_; variable