Home
last modified time | relevance | path

Searched refs:registration (Results 1 – 25 of 222) sorted by relevance

123456789

/external/tensorflow/tensorflow/lite/core/api/
Dop_resolver_test.cc86 const TfLiteRegistration* registration = in TEST() local
88 EXPECT_NE(nullptr, registration); in TEST()
89 EXPECT_EQ(nullptr, registration->init(nullptr, nullptr, 0)); in TEST()
90 EXPECT_EQ(kTfLiteOk, registration->prepare(nullptr, nullptr)); in TEST()
91 EXPECT_EQ(kTfLiteOk, registration->invoke(nullptr, nullptr)); in TEST()
93 registration = resolver->FindOp(BuiltinOperator_CAST, 0); in TEST()
94 EXPECT_EQ(nullptr, registration); in TEST()
96 registration = resolver->FindOp("mock_custom", 0); in TEST()
97 EXPECT_NE(nullptr, registration); in TEST()
98 EXPECT_EQ(nullptr, registration->init(nullptr, nullptr, 0)); in TEST()
[all …]
Dop_resolver.cc22 ErrorReporter* error_reporter, const TfLiteRegistration** registration) { in GetRegistrationFromOpCode() argument
24 *registration = nullptr; in GetRegistrationFromOpCode()
36 *registration = op_resolver.FindOp(builtin_code, version); in GetRegistrationFromOpCode()
37 if (*registration == nullptr) { in GetRegistrationFromOpCode()
49 *registration = op_resolver.FindOp(name, version); in GetRegistrationFromOpCode()
50 if (*registration == nullptr) { in GetRegistrationFromOpCode()
/external/tensorflow/tensorflow/lite/experimental/micro/
Dmicro_mutable_op_resolver_test.cc57 const TfLiteRegistration* registration = in TF_LITE_MICRO_TEST() local
59 TF_LITE_MICRO_EXPECT_NE(nullptr, registration); in TF_LITE_MICRO_TEST()
60 TF_LITE_MICRO_EXPECT_EQ(nullptr, registration->init(nullptr, nullptr, 0)); in TF_LITE_MICRO_TEST()
61 TF_LITE_MICRO_EXPECT_EQ(kTfLiteOk, registration->prepare(nullptr, nullptr)); in TF_LITE_MICRO_TEST()
62 TF_LITE_MICRO_EXPECT_EQ(kTfLiteOk, registration->invoke(nullptr, nullptr)); in TF_LITE_MICRO_TEST()
64 registration = resolver->FindOp(BuiltinOperator_CONV_2D, 10); in TF_LITE_MICRO_TEST()
65 TF_LITE_MICRO_EXPECT_EQ(nullptr, registration); in TF_LITE_MICRO_TEST()
67 registration = resolver->FindOp(BuiltinOperator_RELU, 0); in TF_LITE_MICRO_TEST()
68 TF_LITE_MICRO_EXPECT_EQ(nullptr, registration); in TF_LITE_MICRO_TEST()
70 registration = resolver->FindOp("mock_custom", 0); in TF_LITE_MICRO_TEST()
[all …]
Dmicro_mutable_op_resolver.cc23 const TfLiteRegistration& registration = registrations_[i]; in FindOp() local
24 if ((registration.builtin_code == op) && in FindOp()
25 (registration.version == version)) { in FindOp()
26 return &registration; in FindOp()
35 const TfLiteRegistration& registration = registrations_[i]; in FindOp() local
36 if ((registration.builtin_code == -1) && in FindOp()
37 (strcmp(registration.custom_name, op) == 0) && in FindOp()
38 (registration.version == version)) { in FindOp()
39 return &registration; in FindOp()
46 TfLiteRegistration* registration, in AddBuiltin() argument
[all …]
Dmicro_interpreter.cc42 const char* OpNameFromRegistration(const TfLiteRegistration* registration) { in OpNameFromRegistration() argument
43 if (registration->builtin_code == BuiltinOperator_CUSTOM) { in OpNameFromRegistration()
44 return registration->custom_name; in OpNameFromRegistration()
46 return EnumNameBuiltinOperator(BuiltinOperator(registration->builtin_code)); in OpNameFromRegistration()
174 const TfLiteRegistration* registration = nullptr; in Invoke() local
176 &registration); in Invoke()
180 if (registration == nullptr) { in Invoke()
185 static_cast<BuiltinOperator>(registration->builtin_code); in Invoke()
207 if (registration->builtin_code == BuiltinOperator_CUSTOM) { in Invoke()
215 if (registration->init) { in Invoke()
[all …]
/external/autotest/server/site_tests/firmware_Cr50U2fCommands/
Dfirmware_Cr50U2fCommands.py228 registration = self.__u2f_generate(APP_ID, USER_SECRET_1, '00')
231 if registration['pubKey'] == registration_2['pubKey']:
234 if registration['keyHandle'] == registration_2['keyHandle']:
238 registration = self.__u2f_generate(APP_ID, USER_SECRET_1, '00')
242 self.__u2f_sign(APP_ID, USER_SECRET_1, registration['keyHandle'],
280 registration = self.__u2f_generate(APP_ID, USER_SECRET_1, '00')
287 self.__u2f_sign(APP_ID, USER_SECRET_1, registration['keyHandle'],
291 registration = self.__u2f_generate(APP_ID, USER_SECRET_1, '00')
295 self.__u2f_sign(APP_ID, USER_SECRET_1, registration['keyHandle'],
301 self.__u2f_sign(APP_ID, USER_SECRET_1, registration['keyHandle'],
[all …]
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Dsoftmax_test.cc48 const TfLiteRegistration* registration = in TestSoftmaxFloat() local
50 TF_LITE_MICRO_EXPECT_NE(nullptr, registration); in TestSoftmaxFloat()
56 if (registration->init) { in TestSoftmaxFloat()
57 user_data = registration->init(&context, init_data, init_data_size); in TestSoftmaxFloat()
75 if (registration->prepare) { in TestSoftmaxFloat()
76 TF_LITE_MICRO_EXPECT_EQ(kTfLiteOk, registration->prepare(&context, &node)); in TestSoftmaxFloat()
78 TF_LITE_MICRO_EXPECT_NE(nullptr, registration->invoke); in TestSoftmaxFloat()
79 TF_LITE_MICRO_EXPECT_EQ(kTfLiteOk, registration->invoke(&context, &node)); in TestSoftmaxFloat()
80 if (registration->free) { in TestSoftmaxFloat()
81 registration->free(&context, user_data); in TestSoftmaxFloat()
[all …]
Ddepthwise_conv_test.cc62 const TfLiteRegistration* registration = in TestDepthwiseConvFloat() local
64 TF_LITE_MICRO_EXPECT_NE(nullptr, registration); in TestDepthwiseConvFloat()
75 if (registration->init) { in TestDepthwiseConvFloat()
76 user_data = registration->init(&context, init_data, init_data_size); in TestDepthwiseConvFloat()
94 if (registration->prepare) { in TestDepthwiseConvFloat()
95 TF_LITE_MICRO_EXPECT_EQ(kTfLiteOk, registration->prepare(&context, &node)); in TestDepthwiseConvFloat()
97 TF_LITE_MICRO_EXPECT_NE(nullptr, registration->invoke); in TestDepthwiseConvFloat()
98 TF_LITE_MICRO_EXPECT_EQ(kTfLiteOk, registration->invoke(&context, &node)); in TestDepthwiseConvFloat()
99 if (registration->free) { in TestDepthwiseConvFloat()
100 registration->free(&context, user_data); in TestDepthwiseConvFloat()
[all …]
Dfully_connected_test.cc57 const TfLiteRegistration* registration = in TestFullyConnectedFloat() local
59 TF_LITE_MICRO_EXPECT_NE(nullptr, registration); in TestFullyConnectedFloat()
68 if (registration->init) { in TestFullyConnectedFloat()
69 user_data = registration->init(&context, init_data, init_data_size); in TestFullyConnectedFloat()
87 if (registration->prepare) { in TestFullyConnectedFloat()
88 TF_LITE_MICRO_EXPECT_EQ(kTfLiteOk, registration->prepare(&context, &node)); in TestFullyConnectedFloat()
90 TF_LITE_MICRO_EXPECT_NE(nullptr, registration->invoke); in TestFullyConnectedFloat()
91 TF_LITE_MICRO_EXPECT_EQ(kTfLiteOk, registration->invoke(&context, &node)); in TestFullyConnectedFloat()
92 if (registration->free) { in TestFullyConnectedFloat()
93 registration->free(&context, user_data); in TestFullyConnectedFloat()
[all …]
/external/autotest/client/common_lib/cros/fake_device_server/
Dregistration_tickets_unittest.py40 self.registration = registration_tickets.RegistrationTickets(
50 self.registration.POST, 1234, 'finalize')
58 returned_json = self.registration.POST(1234, 'finalize')
72 'Bearer %s' % self.registration.TEST_ACCESS_TOKEN)
74 returned_json = self.registration.POST()
82 returned_json = self.registration.GET(1234)
87 self.registration.GET, 1235)
101 returned_json = self.registration.PATCH(1234)
116 returned_json = self.registration.PUT(12345)
128 self.registration.PUT, 12345)
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.cc109 const string& device_name, const DeviceRegistration& registration) { in RegisterCompilationDevice() argument
113 registry.compilation_devices_.emplace(device_name, registration); in RegisterCompilationDevice()
115 registration.compilation_device_name); in RegisterCompilationDevice()
132 const string& device_name, const DeviceRegistration** registration) { in GetCompilationDevice() argument
144 DeviceRegistration& registration = in GetCompilationDevice() local
146 registration.compilation_device_name = DEVICE_CPU_XLA_JIT; in GetCompilationDevice()
147 registration.autoclustering_policy = in GetCompilationDevice()
151 registration.compile_all_resource_ops = false; in GetCompilationDevice()
154 DeviceRegistration& registration = in GetCompilationDevice() local
156 registration.compilation_device_name = DEVICE_GPU_XLA_JIT; in GetCompilationDevice()
[all …]
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc28 const TfLiteRegistration& registration, in ReportOpError() argument
32 registration.custom_name in ReportOpError()
33 ? registration.custom_name in ReportOpError()
35 static_cast<BuiltinOperator>(registration.builtin_code)), in ReportOpError()
189 TfLiteContext* context, TfLiteRegistration registration, in ReplaceNodeSubsetsWithDelegateKernels() argument
192 ->ReplaceNodeSubsetsWithDelegateKernels(registration, nodes_to_replace, in ReplaceNodeSubsetsWithDelegateKernels()
271 TfLiteRegistration registration, const TfLiteIntArray* nodes_to_replace, in ReplaceNodeSubsetsWithDelegateKernels() argument
274 registration.builtin_code = BuiltinOperator_DELEGATE; in ReplaceNodeSubsetsWithDelegateKernels()
303 params, &registration, &node_index)); in ReplaceNodeSubsetsWithDelegateKernels()
521 const TfLiteRegistration* registration, int* node_index) { in AddNodeWithParameters() argument
[all …]
/external/tensorflow/tensorflow/lite/profiling/
Dprofile_summarizer_test.cc54 static TfLiteRegistration registration = { in RegisterSimpleOp() local
58 return &registration; in RegisterSimpleOp()
62 static TfLiteRegistration registration = {nullptr, in RegisterSimpleOpWithProfilingDetails() local
70 return &registration; in RegisterSimpleOpWithProfilingDetails()
76 void Init(const std::function<TfLiteRegistration*()>& registration);
90 const std::function<TfLiteRegistration*()>& registration) { in Init() argument
94 SetCustomOp(kOpName, {}, registration); in Init()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_interpreter_device.cc46 XlaOpRegistry::DeviceRegistration registration; in CreateDevices() local
47 registration.compilation_device_name = DEVICE_INTERPRETER_XLA_JIT; in CreateDevices()
48 registration.autoclustering_policy = in CreateDevices()
50 registration.compile_all_resource_ops = true; in CreateDevices()
52 registration); in CreateDevices()
Dxla_cpu_device.cc43 XlaOpRegistry::DeviceRegistration registration; in CreateDevices() local
44 registration.compilation_device_name = DEVICE_CPU_XLA_JIT; in CreateDevices()
45 registration.autoclustering_policy = in CreateDevices()
49 registration.compile_all_resource_ops = true; in CreateDevices()
50 XlaOpRegistry::RegisterCompilationDevice(DEVICE_XLA_CPU, registration); in CreateDevices()
Dmark_for_compilation_pass.cc501 const XlaOpRegistry::DeviceRegistration* registration; in FindCompilationCandidates() local
503 XlaOpRegistry::GetCompilationDevice(device_type.type(), &registration)); in FindCompilationCandidates()
504 DeviceType jit_device_type(registration->compilation_device_name); in FindCompilationCandidates()
506 bool always_auto_cluster = registration->autoclustering_policy == in FindCompilationCandidates()
510 op_filter.allow_resource_ops = registration->compile_all_resource_ops; in FindCompilationCandidates()
545 if (!registration->compile_all_resource_ops && in FindCompilationCandidates()
655 const XlaOpRegistry::DeviceRegistration* registration; in IsCompilable() local
657 &registration)); in IsCompilable()
658 DeviceType jit_device_type(registration->compilation_device_name); in IsCompilable()
715 const XlaOpRegistry::DeviceRegistration* registration; in Run() local
[all …]
Dxla_gpu_device.cc65 XlaOpRegistry::DeviceRegistration registration; in CreateDevices() local
66 registration.compilation_device_name = DEVICE_GPU_XLA_JIT; in CreateDevices()
67 registration.autoclustering_policy = in CreateDevices()
69 registration.compile_all_resource_ops = true; in CreateDevices()
70 XlaOpRegistry::RegisterCompilationDevice(DEVICE_XLA_GPU, registration); in CreateDevices()
/external/tensorflow/tensorflow/lite/experimental/c/
Dc_api_experimental.cc30 const TFL_Registration* registration, in TFL_InterpreterOptionsAddBuiltinOp() argument
34 registration, min_version, max_version); in TFL_InterpreterOptionsAddBuiltinOp()
39 const TFL_Registration* registration, in TFL_InterpreterOptionsAddCustomOp() argument
41 options->op_resolver.AddCustom(name, registration, min_version, max_version); in TFL_InterpreterOptionsAddCustomOp()
/external/tensorflow/tensorflow/python/saved_model/
Drevived_types.py124 for registration in versions:
126 registration.identifier = identifier
127 if registration.version in version_numbers:
130 registration.version, identifier))
131 version_numbers.add(registration.version)
/external/tensorflow/tensorflow/lite/
Dmutable_op_resolver.cc33 const TfLiteRegistration* registration, in AddBuiltin() argument
36 TfLiteRegistration new_registration = *registration; in AddBuiltin()
46 const TfLiteRegistration* registration, in AddCustom() argument
49 TfLiteRegistration new_registration = *registration; in AddCustom()
Dmodel.cc212 const TfLiteRegistration* registration = nullptr; in BuildLocalIndexToRegistrationMapping() local
214 &registration); in BuildLocalIndexToRegistrationMapping()
218 flatbuffer_op_index_to_registration_.push_back(registration); in BuildLocalIndexToRegistrationMapping()
265 const TfLiteRegistration* registration = in ParseNodes() local
267 if (registration == nullptr) { in ParseNodes()
274 static_cast<BuiltinOperator>(registration->builtin_code); in ParseNodes()
287 op->custom_options()->size(), nullptr, registration); in ParseNodes()
295 nullptr, 0, builtin_data, registration); in ParseNodes()
448 for (const auto* registration : flatbuffer_op_index_to_registration_) { in ApplyDelegates() local
449 if ((registration->builtin_code == BuiltinOperator_CUSTOM) && in ApplyDelegates()
[all …]
Dmutable_op_resolver_test.cc30 static TfLiteRegistration registration = { in GetDummyRegistration() local
36 return &registration; in GetDummyRegistration()
44 static TfLiteRegistration registration = { in GetDummy2Registration() local
50 return &registration; in GetDummy2Registration()
/external/autotest/client/site_tests/cellular_DeferredRegistration/
Dcontrol.pseudomodem10 - Shill destroys a service due to temporary registration loss
11 - Shill never destroys a service after permanent registration loss
19 Tests that shill can handle temporary registration loss without destroying
22 still destroys a service that has suffered a registration loss for an
/external/tensorflow/tensorflow/lite/experimental/writer/
Dwriter_lib.cc58 const TfLiteRegistration* registration = &node_and_registration->second; in ExportOperators() local
59 if (!registration->custom_name) { in ExportOperators()
61 GetOpCodeForBuiltin(registration->builtin_code); in ExportOperators()
64 GetOpCodeForCustom(registration->custom_name); in ExportOperators()
72 const TfLiteRegistration& registration = node_and_registration->second; in ExportOperators() local
81 if (!registration.custom_name) { in ExportOperators()
84 fbb, static_cast<enum BuiltinOperator>(registration.builtin_code), in ExportOperators()
89 auto custom_writer = custom_op_to_writer_.find(registration.custom_name); in ExportOperators()
/external/tensorflow/tensorflow/core/lib/monitoring/
Dcollection_registry.cc98 for (const auto& registration : registry_) { in CollectMetrics() local
100 collector.CollectMetricDescriptor(registration.second.metric_def); in CollectMetrics()
103 collector.CollectMetricValues(registration.second /* collection_info */); in CollectMetrics()

123456789