[CODEGEN][CONTRIB] Various update for CoreML codegen - #5934
Conversation
kazum
commented
Jun 26, 2020
- Update relay.ext.coremlcompiler based on the change in [RUNTIME] Introduce MetadataModule to separate code compilation/interpretation and weight initialization #5770.
- Support int32 for Core ML input and output
- Handle "run" in CoreMLRuntime::GetFunction to measure time with Module.time_evaluator.
- Return PackedFunc() when CoreMLRuntime::GetFunction handles nothing.
- Add test for each operator supported by CoreML codegen.
- Support relay.expand_dims and relay.nn.relu.
|
aah, I haven't noticed that CoreML codegen was also using BYOC infra. Thanks for updating it. I have a question which is not relevant to this change. I see |
|
@zhiics Relevant changes to #5770 are:
We can at least pre-allocate memory. I think of doing more optimization for input tensor management in another PR :) |
|
@kazum no problem. We shouldn't do that in this PR. I was just curious if you have a plan to do that. |
|
Thanks @kazum |
* [CODEGEN][CONTRIB] Various update for CoreML codegen * fix lint error
* [CODEGEN][CONTRIB] Various update for CoreML codegen * fix lint error