Searched refs:layer_input (Results 1 – 4 of 4) sorted by relevance
27 const float *layer_input = input + i * axis_dim * inner_dim; in Cumsum() local32 MS_FLOAT32X4 val = MS_LDQ_F32(layer_input + j); in Cumsum()37 *(layer_output + j) = *(layer_input + j); in Cumsum()57 const float *layer_input = input + i * axis_dim * inner_dim + inner_dim * input_offset; in Cumsum() local65 MS_FLOAT32X4 input_val = MS_LDQ_F32(layer_input + k); in Cumsum()73 *(layer_output + k) = *(layer_input + k) + *(layer_last_output + k); in Cumsum()75 layer_input += inner_dim; in Cumsum()87 const float *layer_input = input + i * axis_dim * inner_dim + (axis_dim - 1) * inner_dim; in CumsumReverse() local92 MS_FLOAT32X4 val = MS_LDQ_F32(layer_input + j); in CumsumReverse()97 *(layer_output + j) = *(layer_input + j); in CumsumReverse()[all …]
304 bool layer_input; member
406 info->ps.layer_input = true; in gather_info_input_decl_ps()
4946 if (ps->info.ps.layer_input) { in radv_pipeline_generate_ps_inputs()