Searched refs:PrependDim (Results 1 – 6 of 6) sorted by relevance
85 t2 = t2.PrependDim(3); in TEST_F()86 t2 = t2.PrependDim(2); in TEST_F()87 t2 = t2.PrependDim(1); in TEST_F()101 t2 = t2.PrependDim(-1); in TEST_F()
103 TensorShape PrependDim(dsize_t dim) const;
195 TensorShape TensorShape::PrependDim(dsize_t dim) const { in PrependDim() function in mindspore::dataset::TensorShape
409 shape = shape.PrependDim(size); in ParseNodeFeatureFromMemory()451 shape = shape.PrependDim(size); in ParseEdgeFeatureFromMemory()
414 shape = shape.PrependDim(size); in GetNodeFeature()504 shape = shape.PrependDim(size); in GetEdgeFeature()
191 TensorShape new_shape = first_shape.PrependDim(static_cast<int64_t>(batch_size)); in BatchRows()