Searched refs:begin_params_axis (Results 1 – 1 of 1) sorted by relevance
2208 begin_params_axis=-1, argument2280 if begin_params_axis >= inputs_rank or begin_norm_axis >= inputs_rank:2283 (begin_params_axis, begin_norm_axis, inputs_rank))2284 params_shape = inputs_shape[begin_params_axis:]2288 (inputs.name, begin_params_axis, inputs_shape))