Searched refs:column_shape (Results 1 – 5 of 5) sorted by relevance
40 std::vector<int64_t> column_shape; in LoadFeatureTensor() local42 &col_type_size, &column_shape)); in LoadFeatureTensor()60 std::vector<int64_t> column_shape; in LoadFeatureToSharedMemory() local62 &col_type_size, &column_shape)); in LoadFeatureToSharedMemory()85 std::vector<int64_t> column_shape; in LoadFeatureIndex() local87 &col_type_size, &column_shape)); in LoadFeatureIndex()
76 … uint64_t *column_data_type_size, std::vector<int64_t> *column_shape, in GetColumnTypeByName() argument80 RETURN_UNEXPECTED_IF_NULL(column_shape); in GetColumnTypeByName()90 *column_shape = column_shape_[column_id]; in GetColumnTypeByName()98 std::vector<int64_t> *column_shape) { in GetColumnValueByName() argument101 RETURN_UNEXPECTED_IF_NULL(column_shape); in GetColumnValueByName()109 *column_shape = column_shape_[column_id]; in GetColumnValueByName()
72 std::vector<int64_t> *column_shape);99 uint64_t *column_data_type_size, std::vector<int64_t> *column_shape,
236 std::vector<int64_t> column_shape; in LoadTensorRow() local243 &column_shape, &category)); in LoadTensorRow()263 &column_shape)); in LoadTensorRow()
327 auto column_shape = tensor->shape(); in CheckTensorRowShapes() local329 auto shapes = column_shape.AsVector(); in CheckTensorRowShapes()373 auto column_shape = tensor->shape(); in FetchMetaFromTensorRow() local376 auto shapes = column_shape.AsVector(); in FetchMetaFromTensorRow()