HomeSort by relevance Sort by last modified time
    Searched refs:FindOp (Results 1 - 25 of 35) sorted by null

1 2

  /external/tensorflow/tensorflow/lite/kernels/
register.h 30 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
32 const TfLiteRegistration* FindOp(const char* op, int version) const override;
register_ref.h 30 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
32 const TfLiteRegistration* FindOp(const char* op, int version) const override;
register.cc 151 const TfLiteRegistration* BuiltinOpResolver::FindOp(tflite::BuiltinOperator op,
153 return MutableOpResolver::FindOp(op, version);
156 const TfLiteRegistration* BuiltinOpResolver::FindOp(const char* op,
167 return MutableOpResolver::FindOp(op, version);
register_ref.cc 145 const TfLiteRegistration* BuiltinRefOpResolver::FindOp(
147 return MutableOpResolver::FindOp(op, version);
150 const TfLiteRegistration* BuiltinRefOpResolver::FindOp(const char* op,
161 return MutableOpResolver::FindOp(op, version);
  /external/tensorflow/tensorflow/lite/core/api/
op_resolver.h 30 virtual const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
33 virtual const TfLiteRegistration* FindOp(const char* op,
op_resolver.cc 36 *registration = op_resolver.FindOp(builtin_code, version);
49 *registration = op_resolver.FindOp(name, version);
op_resolver_test.cc 43 const TfLiteRegistration* FindOp(BuiltinOperator op,
53 const TfLiteRegistration* FindOp(const char* op, int version) const override {
87 resolver->FindOp(BuiltinOperator_CONV_2D, 0);
93 registration = resolver->FindOp(BuiltinOperator_CAST, 0);
96 registration = resolver->FindOp("mock_custom", 0);
102 registration = resolver->FindOp("nonexistent_custom", 0);
  /external/tensorflow/tensorflow/lite/experimental/micro/
micro_mutable_op_resolver.h 29 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
31 const TfLiteRegistration* FindOp(const char* op, int version) const override;
micro_mutable_op_resolver_test.cc 58 resolver->FindOp(BuiltinOperator_CONV_2D, 0);
64 registration = resolver->FindOp(BuiltinOperator_CONV_2D, 10);
67 registration = resolver->FindOp(BuiltinOperator_RELU, 0);
70 registration = resolver->FindOp("mock_custom", 0);
76 registration = resolver->FindOp("mock_custom", 10);
79 registration = resolver->FindOp("nonexistent_custom", 0);
micro_mutable_op_resolver.cc 20 const TfLiteRegistration* MicroMutableOpResolver::FindOp(
32 const TfLiteRegistration* MicroMutableOpResolver::FindOp(const char* op,
micro_interpreter_test.cc 48 const TfLiteRegistration* FindOp(BuiltinOperator op,
52 const TfLiteRegistration* FindOp(const char* op, int version) const override {
  /external/tensorflow/tensorflow/lite/tools/optimize/calibration/
logging_op_resolver.h 42 const TfLiteRegistration* FindOp(BuiltinOperator op,
47 const TfLiteRegistration* FindOp(const char* op, int version) const override;
logging_op_resolver.cc 28 base_resolver.FindOp(op_and_version.first, op_and_version.second);
38 const TfLiteRegistration* LoggingOpResolver::FindOp(BuiltinOperator op,
54 const TfLiteRegistration* LoggingOpResolver::FindOp(const char* op,
logging_op_resolver_test.cc 65 auto reg = resolver.FindOp(BuiltinOperator_CONV_2D, 1);
71 reg = resolver.FindOp(BuiltinOperator_ADD, 1);
123 auto reg = resolver.FindOp(BuiltinOperator_CONV_2D, 1);
128 reg = resolver.FindOp(BuiltinOperator_ADD, 1);
  /external/tensorflow/tensorflow/lite/
mutable_op_resolver_test.cc 58 resolver.FindOp(BuiltinOperator_ADD, 1);
70 resolver.FindOp(BuiltinOperator_CONV_2D, 1);
81 found_registration = resolver.FindOp(BuiltinOperator_ADD, 2);
86 found_registration = resolver.FindOp(BuiltinOperator_ADD, 3);
99 found_registration = resolver.FindOp(BuiltinOperator_ADD, 1);
102 found_registration = resolver.FindOp(BuiltinOperator_ADD, 4);
110 const TfLiteRegistration* found_registration = resolver.FindOp("AWESOME", 1);
124 resolver.FindOp("EXCELLENT", 1);
132 const TfLiteRegistration* found_registration = resolver.FindOp("AWESOME", 2);
148 ASSERT_EQ(resolver1.FindOp(BuiltinOperator_ADD, 1)->invoke
    [all...]
mutable_op_resolver.cc 20 const TfLiteRegistration* MutableOpResolver::FindOp(tflite::BuiltinOperator op,
26 const TfLiteRegistration* MutableOpResolver::FindOp(const char* op,
mutable_op_resolver.h 57 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
59 const TfLiteRegistration* FindOp(const char* op, int version) const override;
model_test.cc 58 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
63 const TfLiteRegistration* FindOp(const char* op, int version) const override {
  /external/tensorflow/tensorflow/lite/tools/
verifier.h 29 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
35 const TfLiteRegistration* FindOp(const char* op, int version) const override {
  /external/tensorflow/tensorflow/lite/examples/label_image/
bitmap_helpers_impl.h 60 resolver.FindOp(BuiltinOperator_RESIZE_BILINEAR, 1);
  /external/tensorflow/tensorflow/lite/experimental/writer/
writer_lib_test.cc 44 const TfLiteRegistration* reg = resolver.FindOp(BuiltinOperator_ADD, 1);
  /external/tensorflow/tensorflow/contrib/lite/kernels/
register.cc 132 const TfLiteRegistration* BuiltinOpResolver::FindOp(tflite::BuiltinOperator op,
134 return MutableOpResolver::FindOp(op, version);
137 const TfLiteRegistration* BuiltinOpResolver::FindOp(const char* op,
148 return MutableOpResolver::FindOp(op, version);
  /external/tensorflow/tensorflow/lite/toco/graph_transformations/
identify_dilated_conv.cc 149 model->operators.erase(FindOp(*model, bts_op));
157 model->operators.erase(FindOp(*model, pad_op));
164 model->operators.erase(FindOp(*model, stb_op));
identify_prelu.cc 132 model->operators.erase(FindOp(*model, add_op));
  /external/tensorflow/tensorflow/lite/experimental/micro/kernels/
softmax_test.cc 49 resolver.FindOp(tflite::BuiltinOperator_SOFTMAX, 1);
115 resolver.FindOp(tflite::BuiltinOperator_SOFTMAX, 1);

Completed in 265 milliseconds

1 2