Searched refs:RowPartitionType (Results 1 – 7 of 7) sorted by relevance
24 enum class RowPartitionType { enum34 RowPartitionType row_partition_type) { in RowPartitionTypeToString()36 case RowPartitionType::FIRST_DIM_SIZE: in RowPartitionTypeToString()38 case RowPartitionType::VALUE_ROWIDS: in RowPartitionTypeToString()40 case RowPartitionType::ROW_LENGTHS: in RowPartitionTypeToString()42 case RowPartitionType::ROW_SPLITS: in RowPartitionTypeToString()44 case RowPartitionType::ROW_LIMITS: in RowPartitionTypeToString()46 case RowPartitionType::ROW_STARTS: in RowPartitionTypeToString()53 inline std::vector<RowPartitionType> GetRowPartitionTypesHelper( in GetRowPartitionTypesHelper()56 new std::unordered_map<std::string, RowPartitionType>( in GetRowPartitionTypesHelper()[all …]
27 string RowPartitionTypeToString(RowPartitionType row_partition_type);31 std::vector<RowPartitionType>* row_partition_types);36 ContextType* context, std::vector<RowPartitionType>* row_partition_types) { in GetRowPartitionTypes()46 std::vector<RowPartitionType>* row_partition_types);53 int GetRaggedRank(const std::vector<RowPartitionType>& row_partition_types);
80 std::vector<RowPartitionType> row_partition_types; in TEST()84 ::testing::ElementsAre(RowPartitionType::FIRST_DIM_SIZE, in TEST()85 RowPartitionType::VALUE_ROWIDS, in TEST()86 RowPartitionType::ROW_SPLITS)); in TEST()91 RowPartitionTypeToString(RowPartitionType::FIRST_DIM_SIZE)); in TEST()93 RowPartitionTypeToString(RowPartitionType::VALUE_ROWIDS)); in TEST()95 RowPartitionTypeToString(RowPartitionType::ROW_SPLITS)); in TEST()
29 std::vector<RowPartitionType>* row_partition_types) { in GetRowPartitionTypesHelper()
29 const std::vector<RowPartitionType>& row_partition_types, in ValidateRowPartitionTypesAndShapes()32 for (RowPartitionType row_partition_type : row_partition_types) { in ValidateRowPartitionTypesAndShapes()34 case RowPartitionType::FIRST_DIM_SIZE: in ValidateRowPartitionTypesAndShapes()35 case RowPartitionType::VALUE_ROWIDS: in ValidateRowPartitionTypesAndShapes()36 case RowPartitionType::ROW_SPLITS: 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()54 row_partition_types[1] != RowPartitionType::VALUE_ROWIDS)) { in ValidateRowPartitionTypesAndShapes()57 if (row_partition_types[0] == RowPartitionType::VALUE_ROWIDS) { in ValidateRowPartitionTypesAndShapes()77 if (row_partition_types[i] == RowPartitionType::FIRST_DIM_SIZE) { in ValidateRowPartitionTypesAndShapes()[all …]
69 RowPartitionType GetRowPartitionTypeByDimension(int dimension) { in GetRowPartitionTypeByDimension()70 if (row_partition_types_[0] == RowPartitionType::FIRST_DIM_SIZE) { in GetRowPartitionTypeByDimension()79 if (row_partition_types_[0] == RowPartitionType::FIRST_DIM_SIZE) { in GetRowPartitionTensor()91 case RowPartitionType::VALUE_ROWIDS: in GetMaxWidth()94 case RowPartitionType::ROW_SPLITS: in GetMaxWidth()327 case RowPartitionType::VALUE_ROWIDS: in CalculateOutputIndex()331 case RowPartitionType::ROW_SPLITS: in CalculateOutputIndex()354 const RowPartitionType first_partition_type = row_partition_types_[0]; in GetFirstDimensionSize()356 case RowPartitionType::FIRST_DIM_SIZE: in GetFirstDimensionSize()359 case RowPartitionType::VALUE_ROWIDS: in GetFirstDimensionSize()[all …]
46 std::vector<tensorflow::RowPartitionType> partition_types;49 tensorflow::RowPartitionType GetRowPartitionTypeByDimension( in GetRowPartitionTypeByDimension()52 tensorflow::RowPartitionType::FIRST_DIM_SIZE) { in GetRowPartitionTypeByDimension()63 const tensorflow::RowPartitionType first_partition_type = in GetFirstDimensionSizeT()66 case tensorflow::RowPartitionType::FIRST_DIM_SIZE: in GetFirstDimensionSizeT()68 case tensorflow::RowPartitionType::VALUE_ROWIDS: in GetFirstDimensionSizeT()72 case tensorflow::RowPartitionType::ROW_SPLITS: { in GetFirstDimensionSizeT()146 tensorflow::RowPartitionType::FIRST_DIM_SIZE) { in GetRowPartitionTensor()218 case tensorflow::RowPartitionType::VALUE_ROWIDS: in GetMaxWidth()220 case tensorflow::RowPartitionType::ROW_SPLITS: in GetMaxWidth()[all …]