Searched refs:indices_rank (Results 1 – 4 of 4) sorted by relevance
76 size_t indices_rank = 1 + (data[data_ix++] % kMaxIndicesRank); in FuzzImpl() local79 if (data_ix + indices_rank >= size) { in FuzzImpl()84 for (i = 0; i < indices_rank; i++) { in FuzzImpl()105 for (i = 0; i < indices_rank - 1; i++) { in FuzzImpl()109 int64 last = indices_dims[indices_rank - 1]; in FuzzImpl()
63 const int indices_rank = NumDimensions(indices); in Prepare() local64 const int indices_nd = SizeOfDimension(indices, indices_rank - 1); in Prepare()69 if (indices_rank < 1) { in Prepare()84 const int output_rank = indices_rank + params_rank - indices_nd - 1; in Prepare()87 for (int i = 0; i < indices_rank - 1; ++i) { in Prepare()
133 int64 indices_rank = in XlaGather() local135 dim_numbers.add_offset_dims(i + indices_rank - num_index_dims); in XlaGather()
109 const int64 indices_rank = c->Rank(indices_shape); in DynamicStitchShapeFunction() local118 TF_RETURN_IF_ERROR(c->Subshape(data_shape, indices_rank, &rest)); in DynamicStitchShapeFunction()