Searched refs:registration_ (Results 1 – 6 of 6) sorted by relevance
482 registration_.reset(new XlaOpRegistry::OpRegistration); in XlaOpRegistrationBuilder()483 registration_->name = string(name); in XlaOpRegistrationBuilder()494 registration_->has_device_allowlist = true; in Device()496 registration_->device_allowlist.emplace(device); in Device()503 registration_->has_device_allowlist = true; in Device()504 registration_->device_allowlist.emplace(device); in Device()509 registration_->compilation_only = true; in CompilationOnly()514 registration_->allow_resource_types = true; in AllowResourceTypes()519 registration_->allow_variant_types = true; in AllowVariantTypes()524 registration_->allow_string_type = true; in AllowStringType()[all …]
380 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()
60 const std::unique_ptr<Registration> registration_; variable
128 TfLiteRegistration* registration() { return ®istration_; } in registration()164 TfLiteRegistration registration_; member in tflite::delegates::__anond98bd1180111::MockTfLiteContext
156 : op_(op), registration_(*registration) { in op_()157 registration_.builtin_code = static_cast<int32_t>(op); in op_()158 registration_.version = version; in op_()163 return ®istration_; in FindOp()173 TfLiteRegistration registration_; variable