Searched defs:axis_i (Results 1 – 5 of 5) sorted by relevance
93 const int axis_i = *(reinterpret_cast<const int *>(axis->buffer()) + i); in run_reverse() local
210 int32_t axis_i = axis_data[i]; in Eval() local
1573 int32_t axis_i = axis_attr.getValues<int32_t>()[i]; in matchAndRewrite() local1617 int32_t axis_i = adjusted_axis[i]; in matchAndRewrite()
990 def op_wrapper(input, axis_i, keepdims_i): argument
1343 int64_t axis_i = axis.getInt(); in BuildGatherOp() local