Skip to content
Snippets Groups Projects
Commit 67fe4db0 authored by lixiaoquan's avatar lixiaoquan Committed by Tianqi Chen
Browse files

[NNVM] Fix dtype of output of mean. (#2334)

dtype of count is the same as dtype of inputs[0] when created, but its type may
  change when multiplied by inputs[0]->shape[i]. Which causes dtype of
  output is not same as dtype of input.
parent b076cad5
No related branches found
No related tags found
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment