MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / IsCustomOp

Function IsCustomOp

tensorflow/compiler/mlir/lite/flatbuffer_import.cc:541–545  ·  view source on GitHub ↗

Returns true if this is a custom op.

Source from the content-addressed store, hash-verified

539
540// Returns true if this is a custom op.
541bool IsCustomOp(const std::string& op_name) {
542 return op_name == "tfl.max_pooling_with_argmax_2d" ||
543 op_name == "tfl.max_unpooling_2d" ||
544 op_name == "tfl.convolution_2d_transpose_bias";
545}
546
547// TODO(krzysd) Handle function calls
548StatusOr<Operation*> ConvertOp(

Callers 1

ConvertOpFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected