Searched defs:layer_input (Results 1 – 2 of 2) sorted by relevance
27 const float *layer_input = input + i * axis_dim * inner_dim; in Cumsum() local57 const float *layer_input = input + i * axis_dim * inner_dim + inner_dim * input_offset; in Cumsum() local87 const float *layer_input = input + i * axis_dim * inner_dim + (axis_dim - 1) * inner_dim; in CumsumReverse() local117 … const float *layer_input = input + (i + 1) * axis_dim * inner_dim - 1 - input_offset * inner_dim; in CumsumReverse() local148 const int *layer_input = input + i * axis_dim * inner_dim; in CumsumInt() local178 const int *layer_input = input + i * axis_dim * inner_dim + inner_dim * input_offset; in CumsumInt() local207 const int *layer_input = input + i * axis_dim * inner_dim + (axis_dim - 1) * inner_dim; in CumsumReverseInt() local223 const int *layer_input = input + (i + 1) * axis_dim * inner_dim - 1 - input_offset * inner_dim; in CumsumReverseInt() local
304 bool layer_input; member