Lines Matching refs:sizes
208 /// Manages the dimensions of a Tensor and their sizes.
211 /// \brief Construct a `TensorShape` from the provided sizes.
268 /// Returns sizes of all dimensions.
277 /// Returns true if `*this` and `b` have the same sizes. Ignores
290 /// which case we pad the rest of the sizes with 1.
310 // for the sizes.
435 // sizes.
515 LOG(INFO) << "Sizes: " << sizeof(TensorShape) << " vs "
553 std::vector<int64> sizes;
558 sizes.push_back(sz);
562 s = TensorShape(sizes);
563 sold = TensorShapeOld(sizes);
657 // A few different test cases for tensor sizes for benchmarks
659 std::vector<int64> sizes;
662 sizes = {100};
665 sizes = {100, 1000};
668 sizes = {100, 1000000};
671 sizes = {100, 256, 192, 3};
674 sizes = {1, 2, 1ll << 34, 1, 1, 1};
677 return sizes;
