Searched refs:dims64 (Results 1 – 2 of 2) sorted by relevance
664 std::vector<int64> dims64 = in ScopedTensorDescriptor() local672 std::transform(dims64.cbegin(), dims64.cend(), dims.begin(), in ScopedTensorDescriptor()4254 std::vector<int64> dims64 = in DoPoolBackwardImpl() local4260 for (auto& x : dims64) dest2_size *= x; in DoPoolBackwardImpl()4432 std::vector<int64> dims64 = in DoNormalizeBackwardWithDimensions() local4438 std::transform(dims64.cbegin(), dims64.cend(), dimsint.begin(), in DoNormalizeBackwardWithDimensions()
527 std::vector<int64> dims64 = in CudnnTensorDescriptor() local535 std::transform(dims64.cbegin(), dims64.cend(), dims.begin(), in CudnnTensorDescriptor()