Searched defs:custom_op (Results 1 – 9 of 9) sorted by relevance
30 fn.walk([&](TFL::CustomTfOp custom_op) { in UnwrapTFCustomOps()
91 auto custom_op = builder.create<CustomTfOp>( in runOnFunction() local
82 tflite::BuiltinOperator builtin_op, const char* custom_op) { in AddOperator()
104 tflite::BuiltinOperator builtin_op, const char* custom_op) { in AddOperator()
135 const Operator* custom_op = in TEST_F() local
222 const TfLiteRegistration* FindCustomOp(void* user_data, const char* custom_op, in FindCustomOp()
139 const TfLiteRegistration* MyFindCustomOp(void*, const char* custom_op, in MyFindCustomOp()
147 bool automatic_shape_propagation, bool custom_op, in SimpleDelegate()
1139 if (auto custom_op = dyn_cast<mlir::TFL::CustomOp>(inst)) { in BuildOperator() local1489 if (auto custom_op = dyn_cast<mlir::TFL::CustomTfOp>(inst)) { in BuildSubGraph() local