Searched refs:shape64 (Results 1 – 2 of 2) sorted by relevance
723 absl::Span<const int64> shape64 = pooling_descriptor.window(); in CudnnPoolingDescriptor() local733 std::transform(shape64.cbegin(), shape64.cend(), shape.begin(), in CudnnPoolingDescriptor()
718 absl::Span<const int64> shape64 = pooling_descriptor.window(); in ScopedPoolingDescriptor() local728 std::transform(shape64.cbegin(), shape64.cend(), shape.begin(), in ScopedPoolingDescriptor()