Searched refs:registration_ (Results 1 – 8 of 8) sorted by relevance
39 registration_(registration), in KernelRunner()59 if (registration_.init) { in InitAndPrepare()60 node_.user_data = registration_.init(&context_, init_data, length); in InitAndPrepare()62 if (registration_.prepare) { in InitAndPrepare()63 TF_LITE_ENSURE_STATUS(registration_.prepare(&context_, &node_)); in InitAndPrepare()69 if (registration_.invoke == nullptr) { in Invoke()73 return registration_.invoke(&context_, &node_); in Invoke()
68 const TfLiteRegistration& registration_; variable
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 …]
373 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
127 TfLiteRegistration* registration() { return ®istration_; } in registration()163 TfLiteRegistration registration_; member in tflite::delegates::__anonf41d3e3a0111::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