Searched defs:layer_output (Results 1 – 1 of 1) sorted by relevance
28 float *layer_output = output + i * axis_dim * inner_dim; in Cumsum() local42 float *layer_output = output + i * axis_dim * inner_dim; in Cumsum() local59 float *layer_output = layer_last_output + inner_dim; in Cumsum() local88 float *layer_output = output + i * axis_dim * inner_dim + (axis_dim - 1) * inner_dim; in CumsumReverse() local102 float *layer_output = output + i * axis_dim * inner_dim + (axis_dim - 1) * inner_dim; in CumsumReverse() local119 float *layer_output = layer_last_output - inner_dim; in CumsumReverse() local149 int *layer_output = output + i * axis_dim * inner_dim; in CumsumInt() local163 int *layer_output = output + i * axis_dim * inner_dim; in CumsumInt() local180 int *layer_output = layer_last_output + inner_dim; in CumsumInt() local208 int *layer_output = output + i * axis_dim * inner_dim + (axis_dim - 1) * inner_dim; in CumsumReverseInt() local[all …]