Searched defs:column_shape (Results 1 – 6 of 6) sorted by relevance
583 … const TensorShape &column_shape, const std::string &column_name) { in CreateUint8TensorFromString()708 TensorShape column_shape = TensorShape::CreateUnknownRankShape(); in ParseSingleVarLenColumn() local998 const auto column_shape = TensorShape(shape); in FillAndCopyVarLenString() local1017 const TensorShape column_shape = in MergeDenseVarLenMiniBatches() local1042 auto column_shape = column_descriptor.Shape().InsertDim(0, batch_size); in ParallelParseExample() local1108 auto column_shape = column_descriptor.Shape().InsertDim(0, batch_size); in ParallelParseExample() local
76 … uint64_t *column_data_type_size, std::vector<int64_t> *column_shape, in GetColumnTypeByName()99 std::vector<int64_t> *column_shape) { in GetColumnValueByName()
248 std::vector<int64_t> column_shape; in LoadTensorRow() local
294 std::vector<int64_t> column_shape; in ReadAllAtPageById() local
1670 std::vector<int64_t> column_shape; in GetNext() local
553 auto column_shape = tensor->shape(); in CheckTensorRowShapes() local605 auto column_shape = tensor->shape(); in FetchMetaFromTensorRow() local