Home
last modified time | relevance | path

Searched defs:custom_op (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Draise_custom_ops.cc55 auto custom_op = builder.create<CustomTfOp>( in runOnFunction() local
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions_test.cc135 const Operator* custom_op = in TEST_F() local
/external/tflite-support/tensorflow_lite_support/cc/task/core/
Dtflite_engine.cc222 const TfLiteRegistration* FindCustomOp(void* user_data, const char* custom_op, in FindCustomOp()
/external/tensorflow/tensorflow/lite/c/
Dc_api_experimental_test.cc138 const TfLiteRegistration* MyFindCustomOp(void*, const char* custom_op, in MyFindCustomOp()
/external/tensorflow/tensorflow/lite/delegates/
Ddelegate_test_util.cc133 bool automatic_shape_propagation, bool custom_op) in SimpleDelegate()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc1125 if (auto custom_op = dyn_cast<mlir::TFL::CustomOp>(inst)) { in BuildOperator() local
1456 if (auto custom_op = dyn_cast<mlir::TFL::CustomTfOp>(inst)) { in BuildSubGraph() local
/external/tensorflow/tensorflow/lite/tools/
Dverifier_test.cc104 tflite::BuiltinOperator builtin_op, const char* custom_op) { in AddOperator()