Searched refs:params_rank (Results 1 – 2 of 2) sorted by relevance
62 const int params_rank = NumDimensions(params); in Prepare() local65 if (params_rank < 1) { in Prepare()73 if (indices_nd > params_rank) { in Prepare()84 const int output_rank = indices_rank + params_rank - indices_nd - 1; in Prepare()90 for (int i = indices_nd; i < params_rank; ++i) { in Prepare()
75 params_rank, params_static = _rank(params)79 axes=(list(range(ids_rank, params_rank))81 else math_ops.range(ids_rank, params_rank)))