Searched refs:coords_shape (Results 1 – 2 of 2) sorted by relevance
33 const Shape& coords_shape = coords_array.shape(); in Gather() local42 CHECK_EQ(coords_shape.dims(0), output_array->shape().dims(0)); in Gather()51 CHECK_EQ(stride * coords_shape.dims(0), output_data.size()); in Gather()53 for (int i = 0; i < coords_shape.dims(0); ++i) { in Gather()
28 const RuntimeShape& coords_shape, const CoordsT* coords_data, in Gather() argument38 const int coords_count = coords_shape.FlatSize(); in Gather()