We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ae7191c commit 0405801Copy full SHA for 0405801
src/ngraph_builder.cc
@@ -2238,7 +2238,7 @@ static Status TranslateLogSoftmaxOp(
2238
shared_ptr<ng::Node> ng_inp;
2239
TF_RETURN_IF_ERROR(GetInputNodes(ng_op_map, op, &ng_inp));
2240
auto inp_shape = ng_inp->get_shape();
2241
- int rank = inp_shape.size();
+ size_t rank = inp_shape.size();
2242
auto ng_axis = ng::AxisSet{rank - 1};
2243
// Batch i, class j
2244
// logsoftmax[i, j] = logits[i, j] - log(sum(exp(logits[i])))
0 commit comments