Searched refs:output_shape_rank (Results 1 – 3 of 3) sorted by relevance
235 const int output_shape_rank = ragged_rank + value_shape.DimensionsCount(); in CombineRaggedTensorToTensorShapes() local236 result.Resize(output_shape_rank); in CombineRaggedTensorToTensorShapes()237 for (int i = 0; i < output_shape_rank; ++i) { in CombineRaggedTensorToTensorShapes()
505 int64 output_shape_rank = in InsertDegenerateDims() local508 CHECK_LT(dim_to_insert, output_shape_rank); in InsertDegenerateDims()512 output_shape_dim_bounds.reserve(output_shape_rank); in InsertDegenerateDims()515 for (int64 i = 0; i < output_shape_rank; ++i) { in InsertDegenerateDims()
3228 const int64 output_shape_rank = in ValidateGatherDimensionNumbers() local3233 if (offset_dim < 0 || offset_dim >= output_shape_rank) { in ValidateGatherDimensionNumbers()3238 i, offset_dim, output_shape_rank); in ValidateGatherDimensionNumbers()