Home
last modified time | relevance | path

Searched refs:AddBuiltin (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/
Dregister.cc68 AddBuiltin(BuiltinOperator_RELU, Register_RELU()); in BuiltinOpResolver()
69 AddBuiltin(BuiltinOperator_RELU_N1_TO_1, Register_RELU_N1_TO_1()); in BuiltinOpResolver()
70 AddBuiltin(BuiltinOperator_RELU6, Register_RELU6()); in BuiltinOpResolver()
71 AddBuiltin(BuiltinOperator_TANH, Register_TANH()); in BuiltinOpResolver()
72 AddBuiltin(BuiltinOperator_LOGISTIC, Register_LOGISTIC()); in BuiltinOpResolver()
73 AddBuiltin(BuiltinOperator_AVERAGE_POOL_2D, Register_AVERAGE_POOL_2D()); in BuiltinOpResolver()
74 AddBuiltin(BuiltinOperator_MAX_POOL_2D, Register_MAX_POOL_2D()); in BuiltinOpResolver()
75 AddBuiltin(BuiltinOperator_L2_POOL_2D, Register_L2_POOL_2D()); in BuiltinOpResolver()
76 AddBuiltin(BuiltinOperator_CONV_2D, Register_CONV_2D()); in BuiltinOpResolver()
77 AddBuiltin(BuiltinOperator_DEPTHWISE_CONV_2D, Register_DEPTHWISE_CONV_2D()); in BuiltinOpResolver()
[all …]
Dregister.h31 void AddBuiltin(tflite::BuiltinOperator op, TfLiteRegistration* registration);
/external/tensorflow/tensorflow/contrib/lite/tools/
Dmutable_op_resolver.h45 void AddBuiltin(tflite::BuiltinOperator op, TfLiteRegistration* registration);
Dmutable_op_resolver.cc31 void MutableOpResolver::AddBuiltin(tflite::BuiltinOperator op, in AddBuiltin() function in tflite::MutableOpResolver
Dverifier_test.cc50 resolver_.AddBuiltin(iter, &fake_op_); in TfLiteFlatbufferModelBuilder()