Searched refs:init_i (Results 1 – 2 of 2) sorted by relevance
1110 init_i = (array_ops.shape(flat_elems[0])[0] - 1 if reverse1121 out, init_i + (1 if reverse else 0), initial_accum)1123 loop_in = [init_i, num_elems] + outputs + flat_initial
190 auto init_i = One(builder, S32); in InvertDiagonalBlocks() local191 auto init = Tuple(builder, {init_i, output, scaled_diag_blocks}); in InvertDiagonalBlocks()