Skip to content

Commit 0405801

Browse files
sayantan-nervanaavijit-nervana
authored andcommitted
int to size_t (#82)
1 parent ae7191c commit 0405801

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/ngraph_builder.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2238,7 +2238,7 @@ static Status TranslateLogSoftmaxOp(
22382238
shared_ptr<ng::Node> ng_inp;
22392239
TF_RETURN_IF_ERROR(GetInputNodes(ng_op_map, op, &ng_inp));
22402240
auto inp_shape = ng_inp->get_shape();
2241-
int rank = inp_shape.size();
2241+
size_t rank = inp_shape.size();
22422242
auto ng_axis = ng::AxisSet{rank - 1};
22432243
// Batch i, class j
22442244
// logsoftmax[i, j] = logits[i, j] - log(sum(exp(logits[i])))

0 commit comments

Comments
 (0)