Searched refs:FIRST_DIM_SIZE (Results 1 – 6 of 6) sorted by relevance
25 FIRST_DIM_SIZE, enumerator36 case RowPartitionType::FIRST_DIM_SIZE: in RowPartitionTypeToString()57 {{"FIRST_DIM_SIZE", RowPartitionType::FIRST_DIM_SIZE}, in GetRowPartitionTypesHelper()79 if (row_partition_types[0] == RowPartitionType::FIRST_DIM_SIZE) { in GetRaggedRank()
84 ::testing::ElementsAre(RowPartitionType::FIRST_DIM_SIZE, in TEST()91 RowPartitionTypeToString(RowPartitionType::FIRST_DIM_SIZE)); in TEST()
10 * "FIRST_DIM_SIZE": if value_rowids is used for the first dimension, then it11 is preceeded by "FIRST_DIM_SIZE".78 * "FIRST_DIM_SIZE": if value_rowids is used for the first dimension, then it79 is preceded by "FIRST_DIM_SIZE".
34 case RowPartitionType::FIRST_DIM_SIZE: in ValidateRowPartitionTypesAndShapes()48 if (row_partition_types[i] == RowPartitionType::FIRST_DIM_SIZE) { in ValidateRowPartitionTypesAndShapes()52 if (row_partition_types[0] == RowPartitionType::FIRST_DIM_SIZE && in ValidateRowPartitionTypesAndShapes()77 if (row_partition_types[i] == RowPartitionType::FIRST_DIM_SIZE) { in ValidateRowPartitionTypesAndShapes()
70 if (row_partition_types_[0] == RowPartitionType::FIRST_DIM_SIZE) { in GetRowPartitionTypeByDimension()79 if (row_partition_types_[0] == RowPartitionType::FIRST_DIM_SIZE) { in GetRowPartitionTensor()330 case RowPartitionType::FIRST_DIM_SIZE: in GetFirstDimensionSize()
52 tensorflow::RowPartitionType::FIRST_DIM_SIZE) { in GetRowPartitionTypeByDimension()66 case tensorflow::RowPartitionType::FIRST_DIM_SIZE: in GetFirstDimensionSizeT()146 tensorflow::RowPartitionType::FIRST_DIM_SIZE) { in GetRowPartitionTensor()